Pandas Dataframe to CSV 仅获取第一行

Question

我在 pandas DF 中有以下一组行：

idx	col1	col2	col3	col4
0	A	B	C	D
1	E	F	G	H
1	I	J	K	L
2	M	M	O	P
2	Q	R	S	T

我想将每组索引行转换为 CSV 并打印到文件。

所以我最终得到一个文件，其中一行 idx 0，两行 idx 1，两行 idx 2。

像这样：

文件1
col1,col2,col3,col4
A,B,C,D

文件2
col1,col2,col3,col4
E,F,G,H
I,J,K,L

文件3
col1,col2,col3,col4
M,N,O,P
Q,R,S,T

我有这段代码，但它只给了我每个索引集的第一行：

for i, dfr in Template.TEMPLATE_DF.iterrows(): 
        fpath = path + '\\' + dfr['tmpl.title'].lower().replace(' ', '_') + '_' + str(dfr['tmpl.id']) + '.csv'
        dfr=pd.DataFrame(data=dfr).transpose()
        dfr.to_csv(fpath, sep=',', encoding='utf-8', na_rep='NULL', index=False)

我在这里错过了什么？

Answer 1

这会将每个分组发送到函数，它应该被写入文件。检查 fpath 虽然这会发生变化，因为您不再通过 iterrows 发送一行，而是数据帧的一部分，所以我使用 [0] 来获取 x 的第一行，但就像我说的那样，不确定它是否有效，因为它不是测试。

data='''idx tmpl.id tmpl.title  col3    col4
0   10  title one   C   D
1   20  title two   G   H
1   30  title three K   L
2   40  title four  O   P
2   50  title five  S   T'''

df = pd.read_csv(io.StringIO(data), sep=' \s+', engine='python')
def write_csv(x):
    # print(x['tmpl.title'].iat[0])
    fpath = path + '\\' + x['tmpl.title'].iat[0].lower().replace(' ', '_') + '_' + str(x['tmpl.id'].iat[0]) + '.csv' # this probably isn't correct
    x.transpose()
    x.to_csv(fpath, sep=',', encoding='utf-8', na_rep='NULL', index=False)

# df.groupby(df.index).apply(write_csv)
df.groupby('idx').apply(write_csv)

如果 idx 不是您的索引，则使用

df.groupby('idx').apply(write_csv)