python - 附加来自不同文件的数组并向数据插入一列

Question

在我的情况下，数据是多个文件中可用的二维数组。最后，我想逐行追加这些二维数组。例如，说 fileA 包含

1 2 3
4 5 6

和文件B包含

7 8 9
0 1 2

我希望由此产生的操作是

1 2 3
4 5 6
7 8 9
0 1 2

最简单的方法是按照解决方案 here就像是

data = np.concatenate([np.loadtxt(f) for f in filenames])

但是，我想在此操作期间包含一列。例如，让 x = [3, 5]长度为 x = 文件数。每个文件的行长度是固定的(在本例中为 2)。我希望结果数组为:

3 1 2 3
3 4 5 6
5 7 8 9
5 0 1 2

如何以pythonic的方式实现这一目标？

Answer 1

试试 pd.concat :

pd.concat([pd.read_csv(f, header=None).assign(x=x) 
           for (f,x) in zip(filenames, x_list)]

你会得到 x列最后。如果你坚持先拥有它，你可以尝试:

pd.concat([pd.read_csv(f, header=None) for f in filenames],
          keys=x_list).reset_index(level=0)