numpy:在数据文件中读取并替换关键字

Question

我想在数据文件中读取以下内容

1 key1 3
4 5 key2
7 8 9

并将其保存到数组或列表中。关键字key1和key2应替换为2和6，即最终数组应为

[[1, 2, 3], [4, 5, 6], [7, 8, 9]]

是否有简单的解决方案，例如numpy.genfromtxt或类似的东西？

Answer 1

在这种情况下，仅使用普通python来做：

keys = {'key1': 2, 'key2': 6}
with open('in.txt', 'r') as f:  # Assuming data in 'in.txt'
    arr = [[int(elm) if elm.isnumeric() else keys[elm] for elm in line.split()] for line in f]