大熊猫情节与正值一种颜色和负值另一种

3 回答

• 2

  使用色彩图对分散点进行着色的想法当然是合理的 . 如果您正在使用 plt.scatter 图，则可以提供颜色图在 c 参数中选择颜色的值 .

  这里你只需要两个值，所以 c= np.sign(df.y) 将是一个合适的选择 .

  import matplotlib.pyplot as plt
  import numpy as np
  import pandas as pd
  
  df = pd.DataFrame({'x': np.arange(25), 'y': np.random.normal(0,2500,25)})
  fig, ax = plt.subplots()
  
  ax.scatter(df.x, df.y, c=np.sign(df.y), cmap="bwr")
  
  plt.show()
  

  

  回复于 2024-04-26T19:47:03+08:00
• 0

  使用布尔掩码和where关键字创建2个单独的数据帧 . 条件是if> 0或不是 . 然后逐个绘制两个数据帧，一个在另一个顶部，使用不同的颜色参数 .

  回复于 2024-04-26T19:47:03+08:00
• 0

  拆分数据框并分别绘制它们：

  import matplotlib.pylab as plt
  import numpy as np
  import pandas as pd
  
  df = pd.DataFrame({'x': np.arange(20), 'y': np.random.randn(20)})
  # split dataframes
  df_plus = df[df.y >= 0]
  df_minus = df[df.y < 0]
  print df_plus
  print df_minus
  
  fig = plt.figure()
  ax = fig.add_subplot(1, 1, 1)
  # plot scatter
  ax.scatter(df_plus.x, df_plus.y, color='r')
  ax.scatter(df_minus.x, df_minus.y, color='b')
  ax.autoscale()
  plt.show()
  

  

  如果您希望绘制负数据帧为正写 df.minus.y = -df_minus.y .

  回复于 2024-04-26T19:47:03+08:00