Python中的元素排列

数组“data”的每个元素都必须更改如下：

例如， 4 应该可以在names_A 和data_A 中看到。 4 的名字_A 是“David”。 现在“David”应该出现在names_B 和data_B 中。 “David”的 data_B 是 30。 所以，元素4必须改变30；等等。

import numpy as np

names_A = ['David', 'Mark', 'Brian', 'Michael']
data_A = [4,3,1,2]

names_B = ['Mark', 'David', 'Michael', 'Brian']
data_B = [51,30,11,29]

data = np.array([[4,4,3,3,2,2,1,1,3,3],
                [4,3,3,3,2,2,3,1,3,1],
                [4,2,3,3,2,2,4,1,4,3]])

最简单、最简单的方法是怎样做的？

我尝试如下：

dats = data.ravel()

results = []
for d in dats:
    nam_A = names_A[data_A == int(d)]
    dat_B = data_B[names_B == nam_A]
    results.append(dat_B)

print np.array(results).reshape(data.shape)


[[51 51 51 51 51 51 51 51 51 51]
 [51 51 51 51 51 51 51 51 51 51]
 [51 51 51 51 51 51 51 51 51 51]]

但是，它给出了错误的结果。你会怎么做？

使用字典创建映射。

names_A = ['David', 'Mark', 'Brian', 'Michael']
data_A = [4,3,1,2]

names_B = ['Mark', 'David', 'Michael', 'Brian']
data_B = [51,30,11,29]

lookup_a = dict(zip(names_A, data_A))
lookup_b = dict(zip(names_B, data_B))

mapping = {value_a: lookup_b[key_a] for key_a, value_a in lookup_a.items()}

现在钥匙在mapping将是来自的数字data_A与相应的值data_B.

我从未与numpy但现在看来更换是一件很容易的事。

仅举一个简单列表的例子：

data = [4, 4, 3, 3, 2, 2, 1, 1, 3, 3]
data = [mapping[value] for value in data]

data now is [30, 30, 51, 51, 11, 11, 29, 29, 51, 51].

安装 numpy 后编辑

如果您创建了映射字典，您可以执行以下操作：

data = np.array([[4, 4, 3, 3, 2, 2, 1, 1, 3, 3],
                 [4, 3, 3, 3, 2, 2, 3, 1, 3, 1],
                 [4, 2, 3, 3, 2, 2, 4, 1, 4, 3]])

for row in data:
    for index, value in enumerate(row):
        row[index] = mapping[value]

data is now:

[[32 30 51 51 11 11 29 29 51 51]
 [30 51 51 51 11 11 51 29 51 29]
 [30 11 51 51 11 11 30 29 30 51]]

由于我之前从未使用过 numpy ，可能会有更简单（或更Pythonic）的解决方案，但至少这做了它应该做的事情。