Python oceanPython项目 简单的画二维图【√】

#  https://oceanpython.org/2013/02/25/2d-histogram/
# 画图网址
# %% 这个是一个分割符，相当于MATLAB的%% 
import numpy as np
import matplotlib.pyplot as plt
# %%
# Create some random numbers
n = 100000
x = np.random.randn(n)
y = (1.5 * x) + np.random.randn(n)
# Plot data
fig1 = plt.figure()
plt.plot(x,y,'.r')
plt.xlabel('x')
plt.ylabel('y')
#画第一个红的散点图
plt.savefig('2D_1V1.png',dpi=600)

# Estimate the 2D histogram
nbins = 200 #将x,y分成200列
H, xedges, yedges = np.histogram2d(x,y,bins=nbins)
#二维直方图
# H needs to be rotated and flipped
H = np.rot90(H)#这2布对数据的操作没看懂#矩阵逆时针旋转
H = np.flipud(H)#将行数倒转
# Mask zeros
Hmasked = np.ma.masked_where(H==0,H) # Mask pixels with a value of zero#mask掩模的做法，将H==0的部分掩掉
#H为0时，表示关联的数组无效
## 以上内容表示对数组的一些操作

# Plot 2D histogram using pcolor
fig2 = plt.figure()
plt.pcolormesh(xedges,yedges,Hmasked)#这里面有一些计数的部分

plt.xlabel('x')
plt.ylabel('y')

cbar = plt.colorbar()#colorbar
cbar.ax.set_ylabel('Counts')

plt.savefig('2D_2V1.png',dpi=600)
plt.show()