将x和y标签添加到打印

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

如果您为DataFrame的列和索引添加标签，熊猫将自动提供相应的标签：

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

在这种情况下，您仍然需要手动提供y标签（例如，通过plt.ylabel，如其他答案所示）。

Answer 2

你可以这样使用do it：

import matplotlib.pyplot as plt 
import pandas as pd

plt.figure()
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')
plt.xlabel('xlabel')
plt.ylabel('ylabel')
plt.show()

很明显，你必须用你想要的字符串“x标签”和“y标签”替换它们。

Answer 3

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

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。设置（xlabel=“x标签”，ylabel=“y标签”）。

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