如何过滤数组中不需要的值以进行绘图?ValueError in matplotlib using numpy array

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('Velocitynkm/s')
ax.set_xlabel('Time (min)')

但是，我遇到错误

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

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

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

您的帮助将不胜感激。

Answer 1

在技术人员朋友的帮助下，只需保留此部分即可

解决问题

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

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

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

以及随后的绘图部分。这会将T切成与R相同的维度。