python - 将 x 和 y 标签添加到 Pandas 图

Question

假设我有以下代码，它使用 pandas 绘制了一些非常简单的东西:

import pandas as pd
values = [[1, 2], [2, 5]]
df2 = pd.DataFrame(values, columns=['Type A', 'Type B'], 
                   index=['Index 1', 'Index 2'])
df2.plot(lw=2, colormap='jet', marker='.', markersize=10, 
         title='Video streaming dropout by category')

如何在保留使用特定颜色图的能力的同时轻松设置 x 和 y 标签？我注意到 pandas DataFrames 的 plot() 包装器不采用任何特定的参数。

Answer 1

df.plot() 函数返回一个 matplotlib.axes.AxesSubplot 对象。您可以在该对象上设置标签。

ax = df2.plot(lw=2, colormap='jet', marker='.', markersize=10, title='Video streaming dropout by category')
ax.set_xlabel("x label")
ax.set_ylabel("y label")

或者，更简洁:ax.set(xlabel="x label", ylabel="y label").

或者，索引 x 轴标签会自动设置为索引名称(如果有的话)。所以 df2.index.name = 'x label' 也可以。