绘制数组相同的扩展Matlotlib

Question

我将几个时间序列存储在具有相同扩展名（* .corr.npy）的numpy数组中。 我想用matplotlib在同一图中绘制它。

现在我像这样绘图：

import pylab as plt
import numpy as num
a=num.load('100.corr.npy')
b=num.load('2345.corr.npy')
...
plt.plot(a)
plt.plot(b)
...
plt.savefig('corr', papertype='a4', orientation='portrait', format='ps')

但是由于我有很多数组，所以我想作一个绘图周期。 有人可以帮我吗？

Answer 1

这就是glob标准模块的亮点！ 它将生成与简单格式规则匹配的文件列表。

在您的情况下：

import glob
import numpy as np

array_files = glob.glob('*.corr.npy')
for fname in array_files:
    x = np.load(fname)
    plt.plot(x)

glob.glob将在当前工作目录中运行，因此，您可能想使用绝对路径：

ROOT_DIR = '/some/path/to/array/files/'
array_files = glob.glob(os.path.join(ROOT_DIR, '*.corr.npy'))

---

我看到您使用num作为numpy的别名。 我认为np是numpy别名的实际标准，因此您可以考虑使用它。