如何过滤数组中不需要的值以进行绘图？使用numpy数组的matplotlib中的ValueError

Question

我正在基于OOP的一些代码中处理一个新例程，并在修改数据数组时遇到问题（代码的简短示例如下）。

基本上，这个例程是关于获取数组R，转置它然后对它进行排序，然后过滤掉低于预定值thres的数据。然后，我将这个数组重新转换回原始维度，然后用T的第一个元素绘制每个行。

import numpy as np
import matplotlib.pyplot as plt

R = np.random.rand(3,8)
R = R.transpose() # transpose the random matrix
R = R[R[:,0].argsort()] # sort this matrix
print(R)

T = ([i for i in np.arange(1,9,1.0)],"temps (min)")

thres = float(input("Define the threshold of coherence: "))

if thres >= 0.0 and thres <= 1.0 :
        R = R[R[:, 0] >= thres] # how to filter unwanted values? changing to NaN / zeros ?
else :
        print("The coherence value is absurd or you're not giving a number!")

print("The final results are ")
print(R)
print(R.transpose())
R.transpose() # re-transpose this matrix

ax = plt.subplot2grid( (4,1),(0,0) )
ax.plot(T[0],R[0])
ax.set_ylabel('Coherence')

ax = plt.subplot2grid( (4,1),(1,0) )
ax.plot(T[0],R[1],'.')
ax.set_ylabel('Back-azimuth')

ax = plt.subplot2grid( (4,1),(2,0) )
ax.plot(T[0],R[2],'.')
ax.set_ylabel('Velocity\nkm/s')
ax.set_xlabel('Time (min)')

但是，我遇到了一个错误

ValueError: x and y must have same first dimension, but have shapes (8,) and (3,)

我评论我认为问题可能存在的部分（如何过滤不需要的值？），但问题仍然存在。

如何绘制这两个数组（R和T），同时仍然可以过滤掉低于thres的不需要的值？我可以将这些不需要的值转换为零或NaN，然后​​成功绘制它们吗？如果是的话，我该怎么做？

非常感谢您的帮助。

Answer 1

在技​​术朋友的帮助下，通过保留这一部分可以简单地解决问题

R = R[R[:, 0] >= thres]

因为删除不需要的元素比将它们更改为NaN或零更为可取。然后通过在此部分中添加稍作修改来修复绘图问题

ax.plot(T[0][:len(R[0])],R[0])

以及随后的绘图部分。这将T切成与R相同的尺寸。