如何在Python中将列标题设置为字幕

时间:2017-03-31 08:11:20

标签: python pandas matplotlib plot

我想用简单的命令在Python中绘制几个excel数据图。我目前正在使用以下代码:

import matplotlib.pyplot as plt
fig, axes = plt.subplots(nrows=4, ncols=4, figsize=(12,12))
data_cols = df.columns[8:16]
for data_col, ax in zip(data_cols, axes.ravel()):
   ax.boxplot(df[data_col])

我想将我为data_cols选择的列标题设置为子图的副标题。你知道我该怎么做吗? 提前谢谢。

1 个答案:

答案 0 :(得分:1)

当然,您需要迭代要绘制的轴和列。因此,如果要绘制数据帧的最后8列,还需要迭代最后8个轴。

这样就可以使用列名作为标题ax.set_title(data_col)

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np; np.random.seed(1)

a = np.random.rand(5, 16)
df = pd.DataFrame(a, columns=list("ABCDEFGHIJKLMNOP"))

fig, axes = plt.subplots(nrows=4, ncols=4, figsize=(6,6), sharex=True, sharey=True)
data_cols = df.columns[8:16]
for data_col, ax in zip(data_cols, axes.ravel()[8:16]):
   ax.boxplot(df[data_col])
   ax.set_title(data_col)

plt.tight_layout()   
plt.show()

enter image description here