python - 使用一维作为循环中的迭代器剩余维度迭代 3D numpy

Question

尽管有许多与遍历 3D 数组相关的类似问题，并且在尝试了一些函数(例如 numpy 的 nditer)之后，我仍然对如何实现以下目标感到困惑:

我有一个维度为 (30, 11, 300) 的信号，它对包含 300 个信号点的 11 个信号进行了 30 次试验。

让这个信号用变量x_表示

我有另一个函数，它将 (11, 300) 矩阵作为输入并将其绘制在 1 个图形上(11 个信号包含 300 个信号点绘制在单个图形上)。设此函数为 sliding_window_plot。

目前，我可以让它做到这一点:

x_plot = x_[0,:,:]
for i in range(x_.shape[0]):
    sliding_window_plot(x_plot[:,:])

在 1 个图上绘制相同(第一次试验)11 个信号，包含 300 个点，共 30 次。 我想让它绘制第 i 组信号。不是每次都对信号进行第一次(第 0 次)尝试。有关如何尝试此操作的任何提示？

Answer 1

您应该能够使用 for 循环遍历第一个维度:

for s in x_:
    sliding_window_plot(s)

每次迭代 s 将是下一个形状数组 (11, 300)。