python - 如何避免本节中的重复代码？

Question

ds_trn, ds_val = SetDataset(trn, batch_size, 'train'), SetDataset(val, batch_size, 'val')

如何避免 SetDataset 出现两次，并使其仅出现一次？

标签： python

Answer 1

你可以这样做，但在我看来，它的可读性会降低：

ds_trn, ds_val = (SetDataset(thing, batch_size, name) for thing, name in [(trn, 'train'), (val, 'val')])

这对于两个变量来说看起来有点过头了，但如果你需要一个相对较大的SetDataset对象列表，它真的会大放异彩。