我有一个这样的DataFrame
>>> df
id name score subject
0001 'bob' 100 'math'
0001 'bob' 67 'science'
0001 'bob' 63 'bio'
0002 'jack' 67 'math'
0002 'jack' 98 'science'
0002 'jack' 90 'bio'
0003 'jack' 60 'math'
0003 'jack' 78 'science'
0003 'rose' 87 'bio'
我想将每个id人的数据过滤到一个新的DataFrame中，然后根据其ID写入Excel文件。因此，上述df将被过滤到3个DataFrames它ids是0001，0002并且0003，所有的DataFrames将被写入单独的Excel文件。
首先，获取唯一ID值的列表
uniquevalues = np.unique(df[['id']].values)
然后对其进行迭代，并将每个具有ID的数据框导出到CSV文件中
for id in uniquevalues:
newdf = df[df['id'] == id]
newdf.to_csv("dataframe "+id+".csv", sep='\t')
如果只有这三个ID，则只需传递for和手动执行相同的操作即可，例如
newdf = df[df['id'] == "0001"]
newdf.to_csv("dataframe0001.csv", sep='\t')
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句