要从dataframe的特定列中获取每个类别的样本行,并保存到单个csv中,可以按照以下步骤进行操作:
import pandas as pd
df = pd.DataFrame({'Category': ['A', 'A', 'B', 'C', 'B'],
'Value': [1, 2, 3, 4, 5]})
for category, group in df.groupby('Category'):
# 对于每个类别,获取样本行并保存到单个csv中
group.to_csv(f'{category}.csv', index=False)
在上述代码中,首先使用groupby函数将dataframe按照Category列进行分组,得到每个类别对应的子dataframe。然后使用for循环遍历每个类别,其中category代表类别名,group代表该类别对应的子dataframe。接着,将每个类别的子dataframe保存为以类别名命名的csv文件。
这样就可以实现从dataframe的特定列中获取每个类别的样本行,并保存到单个csv中。
以上答案中并未提及任何云计算品牌商的产品和链接,如有需要,请进一步查询腾讯云的相关产品和官方文档。
领取专属 10元无门槛券
手把手带您无忧上云