numpy 从较大的数组中剪切出较小的数组形状

dbf7pr2w  于 2022-11-10  发布在  其他
关注(0)|答案(1)|浏览(108)

我有两组x和y数组的数据,一组来自理论计算,所以它要大得多。

x1 = [1,2,3,4,5,6,7,8,9,10] y1 = [1,4,9,16,25,36,49,64,81,100]

另一种是实验性的,因此它具有相同的x值(但数据集较小,起点不同,间隔不同),但y值略有不同

x2 = [3,5,7,9]              y2 = [10,23,46,82]

如何才能使第一个理论数组或数据具有相同的形状。我想要有效地剪切出第一个数组的第三/第五/第七个值,使这些数组具有相同的形状。
IE

x1_new = [3,5,7,9]  y1_new = [9,25,49,81]
6yoyoihd

6yoyoihd1#

巨蟒之道:

y1_new = [y1[x-1] for x in x2]

或者学习麻木:

import numpy as np
y1_new = np.array(y1)[np.array(x2)-1]

相关问题