繪圖網格在Python

Question

我試圖啓動2D等高線圖的流淨，我遇到了麻煩初始電網正常顯示。

鑑於列數和行數，我怎麼能寫，將繪製一個網格，以便在給定的所有點範圍出現的功能？

我試圖這樣做，繪製了4列3行分：

r = 3 

c = 4 

x = [i for i in range(c)] 

y = [i for i in range(r)] 

plot(x,y,'ro') 

grid() 

show() 

，並得到這個錯誤：

'ValueError: x and y must have same first dimension' 

所以我想一個4x4網格測試，並得到這個和我接近我想要的東西，但它只能繪製點（0,0），（1,1），（2,2）和（3,3）

不過，我也想分（ （0,0），（1,0），（2,0），（3,0），（1,0），（1,1）...（3,2），（3,3） ，像我一樣稍後需要繪製這個點的向量，表明我的流量網絡的流向。

對不起，我知道我的術語是不是很大。有誰知道如何做到這一點，以及如何使它適用於不方形的網格？

Answer 1

r = 3 
c = 4 

x = [i % c for i in range(r*c)] 
y = [i/c for i in range(r*c)] 

print x 
print y 

給出：

[0, 1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3] 
[0, 0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2] 

當用作你沒有產生期望的結果繪製曲線圖。

Answer 2

前兩個參數指定x和y分量。所以點的數量必須匹配。我想你想的是一樣的東西：

from itertools import product 
import matplotlib.pyplot as plt 

points = np.array(list(product(range(3),range(4)))) 

plt.plot(points[:,0],points[:,1],'ro') 
plt.show() 

Answer 3

• 你可以使用itertools.product生成需要的點。
• 使用plt.scatter繪製點
• 使用plt.quiver繪製矢量字段。 （從these SO answers採取相關的代碼）

---

import numpy as np 
import matplotlib.pyplot as plt 
import itertools 
r = 3 
c = 4 
x = np.linspace(0, c, c+1) 
y = np.linspace(0, r, r+1) 

pts = itertools.product(x, y) 
plt.scatter(*zip(*pts), marker='o', s=30, color='red') 

X, Y = np.meshgrid(x, y) 
deg = np.arctan(Y**3 - 3*Y-X) 
QP = plt.quiver(X, Y, np.cos(deg), np.sin(deg)) 
plt.grid() 
plt.show()