如何根据csv标题（列）将数据拆分为两个或多个csv文件

Question

嘿，伙计们，我有一个程序，它读取csv文件，随机生成数据并将其转储到另一个csv文件中...现在我有一个csv文件，其中有100列，其中30列属于其他一些数据，但在最后生成时，我想通过附加一些对csv文件通用的公共列将这些数据分割成两个不同的csv文件。这是csv示例数据：

A1  A2  A3  A4  B1  B2  B3  C1  C2  C3
1   1   1   1   2   2   2   3   3   3
1   1   1   1   2   2   2   3   3   3
1   1   1   1   2   2   2   3   3   3

在这种情况下，两个文件都有一个共同点，因此输出数据应该如下所示：

A1  A2  A3  A4  B1  B2  B3
1   1   1   1   2   2   2
1   1   1   1   2   2   2
1   1   1   1   2   2   2


A1  A2  A3  A4  C1  C2  C3
1   1   1   1   3    3   3
1   1   1   1   3    3   3
1   1   1   1   3    3   3

为了得到一个带有所有条件的csv文件，我写了一个代码，这是它的小提琴链接：https://pyfiddle.io/fiddle/6c547de1-eca3-4e86-bf85-bfa9eba6e653/?m=Uploaded文件

这是复杂的理解所以在简单的：我正在读一个csv文件：

df=pd.read_csv("somefile.csv")

在这里执行所有计算操作，并在数据生成时添加另外三列srid，年，月，所以年和月，最后添加srid...final_Array.append（值）#"final_Array"包含所有列的数据。

data = pd.DataFrame(final_Array) to dump it to the csv file .....

但我不知道如何将这些数据值分割成两个文件，所有的操作都是将所有的数据分割成一个文件，有没有办法将写入csv文件之前已经生成的数据进行分割。。。谢谢你的帮助谢谢

Answer 1

您可以使用startswithwith~运算符放弃列，

A_C = df1[df1.columns[~df1.columns.str.startswith('B')]]

    A1  A2  A3  A4  C1  C2  C3
0   1   1   1   1   3   3   3
1   1   1   1   1   3   3   3
2   1   1   1   1   3   3   3


A_B = df1[df1.columns[~df1.columns.str.startswith('C')]]


    A1  A2  A3  A4  B1  B2  B3
0   1   1   1   1   2   2   2
1   1   1   1   1   2   2   2
2   1   1   1   1   2   2   2

如果您想直接将数据帧写入csv，请尝试

df1[df1.columns[~df1.columns.str.startswith('C')]].to_csv('A_B.csv', index = False)