如何根据Y轴的值更改图表数据点的颜色？(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()，以便每个列都有x值(A、B、C、D)，y是实际值。然后为每一行和每一幅图指定一个颜色。更新后的代码如下。

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语句中的“灰色if 0”，这就是我的完美解决方案：

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))

图表