首页 > 解决方案 > 是否可以制作一次只包含一行 csv 的数据框?

问题描述

我正在尝试为我拥有的 csv 文件中的每一行制作条形图。我可以一次性为每一行制作条形图,但我想从每个条形图中制作一个 png 文件,因此一次用一行填充数据框会非常方便。

这是我的一段代码,它可以从我的 CSV 文件中一次性制作条形图:

df2 = df.set_index('ID').T
df2.plot(kind='bar', subplots=True, layout=(2,2), legend=False, color='g')
plt.show()

这会为每一行生成条形图,但我希望能够将它们分开并为每个图表制作一个图像文件

标签: pythonpandasdataframe

解决方案


您可以遍历每一行,将其转换为 DataFrame 并为该行创建条形图

#Iterate through each row
for index, row in df.iterrows():
   
    #Convert row to dataframe
    df_row = row.to_frame().T

    #Proceed as before
    df2 = df_row.set_index('ID').T
    df2.plot(kind='bar', subplots=True, layout=(2,2), legend=False, color='g')

推荐阅读