python - 在文本文件中堆叠多列

Question

我正在尝试通过numpy.savetxt. 数据应该组织在我从二维列表中读取的列中：基本上我需要写

np.column_stack((noise[:,0], noise[:,1], ..., noise[49]))

最后应该有 50 列。

有什么方法可以避免手动编写并自动执行（甚至可能针对不同数量的列）？

标签： pythonnumpymultiple-columns

Answer 1

How about list or generator comprehension?

np.column_stack((noise[:, idx] for idx in range(50)))
np.column_stack(tuple(noise[:, idx] for idx in range(50)))
np.column_stack([noise[:, idx] for idx in range(50)])