【发布时间】:2016-09-29 20:00:22
【问题描述】:
我需要在 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)
【问题讨论】:
标签: python r pandas scatter-plot subplot