python - 如何解决六角形图中的边缘问题？

Question

我是初学者，希望能把问题暴露的清楚一些。

我创建了一个这样的矩阵:

 [0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0]
 [0 0 6 8 9 1 0 0]
 [0 0 4 6 5 4 0 0]
 [0 0 4 2 8 9 0 0]
 [0 0 1 3 6 7 0 0]
 [0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0]

重点是我必须创建一个六角形图，其中色标代表各个单元格中的随机数。这就是我所做的:

import numpy as np
import matplotlib.pyplot as plt 
n=4

A=np.zeros([2*n,2*n], dtype=int)
B=np.random.randint(1,10, size=(n,n))

A[2:6,2:6]=B


plt.figure(figsize=(5,5))
plt.imshow(A, origin=['lower'], cmap=plt.cm.Purples_r)
plt.colorbar()


x=[]
y=[]
for i in range (np.shape(A)[0]):
    for j in range (np.shape(A)[1]):
        N_occurence=A[i,j]
        print(N_occurence)
        for k in range (N_occurence):
            x=np.append(x, i)
            y=np.append(y, j)

plt.figure(figsize=(5,5))
plt.hexbin(x,y,gridsize=(10), cmap=plt.cm.Purples_r)
plt.xlim([1, 6])
plt.ylim([1, 6])
plt.colorbar()

plt.show()

但我无法解决边缘问题，我总是得到半个六边形并且绘图不准确。有谁知道更简单的方法或类似的例子吗？

Answer 1

我仍然不太确定，你在找什么，但我想你想要一个使用像 hexbin 这样的六边形的 imshow 绘图？

也许这有一点帮助:

import matplotlib.pyplot as plt
import numpy as np

# Generate array
A = np.zeros([8, 8], dtype=int)
A[2:6, 2:6] = np.random.randint(1, 10, size=(4, 4))

# Print array
print(A)

# `imshow` plot
plt.figure(figsize=(5,5))
plt.imshow(A, extent=(0, 8, 0, 8), origin='lower')
plt.colorbar()

# Rewrite array to get x and y values
# TODO: There has to be a better way than to use two `for` loops
X = []
Y = []
for y in range(len(A)):
    for x, n in enumerate(A[len(A)-y-1]):
        X += [x]*n
        Y += [y]*n

# `scatter` plot to visualize rewritten array data
plt.figure(figsize=(5,5))
plt.scatter(X, Y)

# `hexbin` plot
plt.figure(figsize=(5,5))
plt.hexbin(X, Y, gridsize=5, extent=(0, 7, 0, 7))
plt.colorbar()

# show plots
plt.show()

随机数组A的结果

[[0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0]
 [0 0 3 7 3 3 0 0]
 [0 0 3 5 8 1 0 0]
 [0 0 4 8 7 3 0 0]
 [0 0 1 7 9 3 0 0]
 [0 0 0 0 0 0 0 0]
 [0 0 0 0 0 0 0 0]]

在

imshow

分散

hexbin

我认为您最好使用自定义解决方案，例如使用指定颜色绘制六边形方 block 的散点图。