如何根據 Y 軸的值更改圖表數據點的 colors？ （Python：Matplotlib）

Question

我的目標是以紅色顯示負數據點（與 Y 軸相關），那些為零的值以灰色顯示，只有正值以綠色顯示

這是我的問題的簡化數據集：

這是我當前的代碼：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

data = pd.read_excel('data.xlsx')
plt.plot([data.A, data.B,data.C, data.D], 'o', c= 'g')
plt.xticks([0,1,2,3], ['A', 'B', 'C', 'D'])
plt.yticks(np.arange(-4, 6, step=2))

我希望有人可以幫助我。

非常感謝！

Answer 1

要為每個條目分配顏色，您需要首先stack() dataframe 以便每列將具有 x 值（A，B，C，D）和 y 將是實際值。 然后為每一行和 plot 分配一個顏色。 更新后的代碼如下。

df=data.stack().reset_index().drop(columns=['level_0']) #Stack data and remove unnecessary column
df.rename(columns={'level_1':'x', 0:'val'}, inplace=True) #Rename for easy identification
df['color'] = ["green" if i > 0 else "red" for i in df['val']] #Create color column
plt.scatter(df.x, y=df.val, c= df.color) #Plot with color = color column
plt.xticks([0,1,2,3], ['A', 'B', 'C', 'D'])
plt.yticks(np.arange(-4, 6, step=2))

Answer 2

謝謝邁克爾，

我只需要使用“如果為零則為灰色”來調整 if 語句，這是我的完美解決方案：

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

data = pd.read_excel('data.xlsx')

df=data.stack().reset_index().drop(columns=['level_0']) #Stack data and 
#remove unnecessary column
df.rename(columns={'level_1':'x', 0:'val'}, inplace=True) #Rename for 
# easy identification
df['color'] = ["green" if i > 0 else ("grey" if i == 0 else "red") for i 
in df['val']] #Create color column
plt.scatter(df.x, y=df.val, c= df.color) #Plot with color = color column
plt.xticks([0,1,2,3], ['A', 'B', 'C', 'D'])
plt.yticks(np.arange(-4, 6, step=2))

解決方案_圖表