我想知道有一个" easy"创建两个匹配数组的方法,在两个数组中插入一些虚拟缺失值,使它们保持相同的大小,两个数组中相同的索引保持不变,例如:
["A", "B", "C", "D", "E", "F"]
和["B", "C", "E"]
将是
["A", "B", "C", "D", "E", "F"]
和["N/A", "B", "C", "N/A", "E", "N/A"]
提前致谢: - )
答案 0 :(得分:7)
列表理解中的单行代码:
array_1 = ["A", "B", "C", "D", "E", "F"]
array_2 = {"B", "C", "E"}
array_3 = [x if x in array_2 else "N/A" for x in array_1]
print(array_3)
结果:
['N/A', 'B', 'C', 'N/A', 'E', 'N/A']
请注意,我将array_2
转换为set
以便更快地查找。
答案 1 :(得分:3)
array_1 = ["A", "B", "C", "D", "E", "F"]
array_2 = ["B", "C", "E"]
array_3 = array_1
for n,x in enumerate(array_3):
if x not in array_2:
array_3[n] = np.nan
print (array_3)
出:
[nan, 'B', 'C', nan, 'E', nan]