Python 将列表元素与numpy数组中的列进行比较
我有一个python列表,如下所示Python 将列表元素与numpy数组中的列进行比较,python,numpy,Python,Numpy,我有一个python列表,如下所示 list=['M', 'R', 'W'] array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]] new_array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4]] 和一个numpy数组,如下所示 list=['M', 'R', 'W'] array=[[
list=['M', 'R', 'W']
array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]
new_array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4]]
和一个numpy数组,如下所示
list=['M', 'R', 'W']
array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]
new_array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4]]
我想将此列表中的每个元素
与数组
的第一列进行比较,然后使用匹配的元素创建一个新数组
。因此,典型的输出如下所示
list=['M', 'R', 'W']
array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]
new_array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4]]
我尝试了以下代码
new_array=np.empty((4,3))
for i in range (0,len(list)):
if list[i]==array[i; 0:1]
new_array=np.append(new_array, (array[i,1:4].reshape(4,3)), axis=0)
这真的是一个numpy数组,还是只是一个名称为“数组”的数组
In [238]: np.array([['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]
...: ])
Out[238]:
array([['M', '360.0', '360.0'],
['R', '135.9', '360.0'],
['W', '101.4', '-125.4'],
['Y', '115.8', '-160.4']], dtype='<U6')
列表理解很好地完成了这项工作:
In [243]: [item for item in alist if item[0] in list1]
Out[243]: [['M', 360.0, 360.0], ['R', 135.9, 360.0], ['W', 101.4, -125.4]]
请执行以下操作:
list=['M', 'R', 'W']
array=[['M',360.0, 360.0],['R', 135.9, 360.0],['W', 101.4, -125.4], ['Y', 115.8, -160.4]]
new_array = [x for x in array if x[0] in list]
print(new_array)
新的_数组=[i表示_数组中的i,如果在_列表中的i[0]