原文地址

对 Numpy 矩阵进行直方图数值统计

说实话对矩阵进行可视化在深度学习中需求很大,例如可视化权重分布。

np.ravel 将矩阵展开,plt.hist 绘制直方图。

import numpy as np
import matplotlib.pyplot as plt

a = np.random.randn(1000, 1000)
plt.hist(a.ravel(), bins=np.arange(np.min(a), np.max(a), 0.01))
plt.show()