如何更改列表中的元素位置

Question

我有一個置換矩陣，它是一個方陣，元素是1或0。

p = [[1, 0, 0, 0],
     [0, 0, 1, 0],
     [0, 1, 0, 0],
     [0, 0, 0, 1]]

如果我將p乘以int列表，則列表中元素的位置將被更改。 例如

a_num = [1, 3, 2, 4]
np.dot(np.array(p), np.array(a_num).reshape(4,1))
# results is [1, 2, 3, 4]

現在我要更改str的列表：

a_str = ['c1', 'c3', 'c2', 'c4']

至

['c1', 'c2', 'c3', 'c4']

你知道如何用矩陣p實現它嗎？ 請注意，我的真實應用程序可以在列表中包含數十個元素。

供你參考。 有一篇關於如何根據兩個str列表生成置換矩陣的帖子

Answer 1

你可以使用numpy.take ：

>>> numpy.take(numpy.array(a_str), numpy.array(a_num)-1)
array(['c1', 'c2', 'c3', 'c4'], dtype='|S2')

Answer 2

你可以通過在列表理解中利用枚舉來實現它而不是numpy：

p = [[1, 0, 0, 0],
     [0, 0, 1, 0],
     [0, 1, 0, 0],
     [0, 0, 0, 1]]

a_str = ['c1', 'c3', 'c2', 'c4']

b_str = [a_str[idx] for row in p for idx,i in enumerate(row) if i == 1]

print(b_str)

輸出：

 ['c1', 'c2', 'c3', 'c4']

它接受p每個內部列表，並在該內部列表的idx處使用a_str的元素作為a_str ，由此創建一個新列表。