python - Pandas scatter_matrix 模拟函数对(lower.panel，upper.panel)

Question

我需要在 Python 中创建一个散布矩阵。我尝试使用 scatter_matrix 来实现此目的，但我只想保留对角线上方的散点图。

我正处于真正的开始阶段(没有走得太远)，当列有名称(不是默认数字)时我会遇到麻烦。

这是我的代码:

import itertools
import numpy as np
import matplotlib.pyplot as plt
import pandas as pd

data=pd.DataFrame(np.random.randint(0,100,size=(10, 5)), columns=list('ABCDE')) #THE PROBLEM IS HERE - I WILL HAVE COLUMNS WITH NAMES

d = data.shape[1]

fig, axes = plt.subplots(nrows=d, ncols=d, sharex=True, sharey=True)
for i in range(d):
    for j in range(d):
            ax = axes[i,j]
            if i == j:
                ax.text(0.5, 0.5, "Diagonal", transform=ax.transAxes,
                        horizontalalignment='center', verticalalignment='center',
                        fontsize=16)
            else:
                ax.scatter(data[j], data[i], s=10)

Answer 1

从数据框中选择列时遇到问题。您可以使用 iloc 根据整数位置选择列。将最后一行更改为:

ax.scatter(data.iloc[:,j], data.iloc[:,i], s=10)

给予: