python - 如何绘制 Pandas 数据框的多列

Question

我需要使用波士顿住房数据集制作散点图。我想用 MEDV 列绘制所有其他列。此代码将所有绘图绘制在同一张图上。我怎样才能把它们分开？ enter image description here

import matplotlib.pyplot as plt
%matplotlib inline
fig, axes = plt.subplots(nrows=3, ncols=2, figsize=(12, 12))
for column, ax in zip(['CRIM', 'ZN','INDUS', 'CHAS', 'NOX', 'RM'], axes):
    plt.scatter(boston_df[column], boston_df.MEDV)

Answer 1

如果您展平 axes 对象，您的代码将起作用，因为当前您在 axes 上循环一次，这是一个二维对象。因此，在 for 循环中使用 axes.flatten()，然后使用 ax.scatter 将每一列绘制成一个新图形。

绘制的顺序会是第一行，第二行，第三行

fig, axes = plt.subplots(nrows=3, ncols=2, figsize=(12, 12))
for column, ax in zip(['CRIM', 'ZN','INDUS', 'CHAS', 'NOX', 'RM'], axes.flatten()):
    ax.scatter(boston_df[column], boston_df.MEDV)