3

我有 6 个列表，我想为所有可能的组合创建散点图。这意味着我想创建 n(n-1)/2 个组合，所以 15 个图。我已根据以下脚本正确完成此操作。

for i in d:
    for j in d:
        if(j>i):
            plt.cla()   # Clear axis
            plt.clf()   # Clear figure
            correlation_coefficient = str(np.corrcoef(d[i], d[j])[0][1])
            plt.scatter(d[i],d[j])
            plt.xlabel(names[i])
            plt.ylabel(names[j])
            plt.title('Correlation Coefficient: '+correlation_coefficient)
            plt.grid()
            plt.savefig(names[i]+"_"+names[j]+".png")

我想使用子图将所有这些图保存在一个图中，其中第一行将具有组合 (0,1) (0,2) (0,3) (0,4) (0,5) 第二行 ( 1,2) (1,3) (1,4) (1,5) 第三排 (2,3) (2,4) (2,5) 等

所以最终的结果将是一个包含三角形子图的图形。

更新：

如果我使用子图（下面的代码），我能够以某种方式得到结果，但它不是最佳的，因为我创建了一个 6x6 帧，而你可以用 5x5 来做到这一点。

fig = plt.figure()
cnt = 0

# Create scatterplots for all pairs
for i in d:
    for j in d:
        if(i>=j):
            cnt=cnt+1
        if(j>i):
            cnt += 1
            fig.add_subplot(6,6,cnt)   #top left
            correlation_coefficient = str(np.corrcoef(d[i], d[j])[0][1])
            plt.scatter(np.log(d[i]),np.log(d[j]))

fig.savefig('test.png')
4

1 回答 1

0

使用网格规范：

from matplotlib import pyplot as plt

fig = plt.figure()

data = [(1,2,3),(8,2,3),(0,5,2),(4,7,1),(9,5,2),(8,8,8)]
plotz = len(data)
for i in range(plotz-1):
    for j in range(plotz):
        if(j>i) :
            print(i,j)
            ax = plt.subplot2grid((plotz-1, plotz-1), (i,j-1))
            ax.xaxis.set_ticklabels([])
            ax.yaxis.set_ticklabels([])
            plt.scatter(data[i],data[j]) # might be nice with shared axis limits

fig.show()

来自 6 元素列表的非冗余组合图

使用 add_subplot，您遇到了一个从 MATLAB 继承的奇怪现象，它对子图计数进行了 1 索引。（您也有一些计数错误。）这是一个仅跟踪各种索引的示例：

from matplotlib import pyplot as plt

fig = plt.figure()
count = 0

data = [(1,2,3),(8,2,3),(0,5,2),(4,7,1),(9,5,2),(8,8,8)]
plotz = len(data)
for i in range(plotz-1):
    for j in range(plotz):
        if(j>i):
            print(count, i,j, count -i)
            ax = fig.add_subplot(plotz-1, plotz-1, count-i)
            ax.xaxis.set_ticklabels([])
            ax.yaxis.set_ticklabels([])
            plt.text(.15, .5,'i %d, j %d, c %d'%(i,j,count))
        count += 1

fig.show()

注意：做明显的错误（你的原始代码add_subplot(5,5,cnt))是一个很好的提示：

...用户/lib/python2.7/site-packages/matplotlib/axes.pyc in init (self, fig, *args, **kwargs)

第9249章

9250 其他：

-> 9251 self._subplotspec = GridSpec(rows, cols)[int(num) - 1]

9252 # num - 1 用于从 MATLAB 转换为 python 索引

于 2014-05-09T02:36:18.670 回答