如何在一個 plot 中 CSV 文件中的 plot 多個值？

Question

首先請不要判斷我的英語我不是本地人。

我有一個大的 CSV 文件，我有一個 84000 列和 5 行如下：

filename    max     probability  size_start size_end
   file1     33     0.001         30          10
   file2     4      0.001         30          10
   file3     50     0.001         40          10
   file4     0      0.001         50          10

我想針對每對size_start and start end的max plot probability ，我有 10 對不同的開始和結束。
我想我必須做一個 for 循環或 while 循環來循環開始和結束以及 plot 對應的概率值和最大值，但我不知道如何循環 csv 文件。 如果您對答案有任何想法，請賜教。

Answer 1

在將數據細分為組之后，您可以使用 Panda 的DataFrame.plot()方法：

import pandas as pd

df = pd.read_csv('data.csv')
for (size_start, size_end), group in df.groupby(['size_start', 'size_end']):
    ax = group.plot(x='probability', y='max', kind='scatter')
    ax.get_figure().savefig(f'plot_{size_start}_{size_end}.png')

這將為每個 size_start/size_end 組合創建一個單獨的圖像。 有很多選項可以自定義 plot 的設計和布局，請查看文檔。