Python:用0s 1s H键数据绘制数据

Question

我有一个分子动力学模拟数据。该体系有254个溶质分子和近12000个水分子。仿真有近4700帧。我提取了H键数据。数据显示，如果任何一个溶质分子与水分子都有H键，它就会显示1，否则0。我要绘制H键数据。总共有254*4700个数据点。数据与给定的示例类似。

S1 S2 S3 S4 S5 ...
0  0  0  0  0  ...
0  0  0  0  0  ...
0  1  1  0  0  ...
0  0  0  0  0  ...
0  0  1  1  1  ...
0  0  0  0  1  ...
0  1  0  0  1  ...
0  0  0  0  1  ...
...

我想要像如果数据点是1，它显示一个颜色，否则，如果0，没有颜色(就像任何其他的绘图，例如散点图)。此外，我希望在图中有两个轴，使x轴=溶质数(1 . 254) y轴=帧数(1 . 4700)

因此，在y轴上，只有与x轴相关的数据点应该是有1的颜色。

任何帮助都将不胜感激。非常感谢！

Answer 1

我建议plt.imshow来完成这项任务：

import matplotlib.pyplot as plt
import numpy as np

solutes = 254
frames = 4700
data = np.round(np.random.rand(frames, solutes))
plt.imshow(data, aspect='auto', interpolation='none')
plt.show()

​

​