替换numpy ndarray中的值

Question

我有一个非常大的列表列表，其中包含 numpy ndarrays，我需要将字母映射到整数值。

这与我认为可能有效的方式一致，但它似乎并没有捕获所有数组。

import numpy as np

x = [np.array(['a','b','c']),np.array(['d','e']),np.array(['a','e'])]
dict_x = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e':5}

x[ x == 'a'] = dict_x.get('a')

Output: [1, array(['d', 'e'], 
  dtype='<U1'), array(['a', 'e'], 
  dtype='<U1')]

我还尝试遍历所有键以将它们一一替换为以下内容

for i in dict_x.keys():
    x[ x == i] = dict_x.get(i)

但这会返回，我认为这是有道理的。任何人都有任何聪明的方法来一次和在所有情况下替换所有这些值？非常感谢！

[5, array(['d', 'e'], 
  dtype='<U1'), array(['a', 'e'], 
  dtype='<U1')]

Answer 1

你可以试试这个：

x = [np.array(['a','b','c'], dtype="<U4"),np.array(['d','e'], dtype="<U4"),np.array(['a','e'], dtype="<U4")]
dict_x = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e':5}

for i in x:
    i[ i == 'a'] = dict_x.get('a')

x = array(['1', 'b', 'c'],
      dtype='<U1'), array(['d', 'e'],
      dtype='<U1'), array(['1', 'e'],
      dtype='<U1')]