Python：从多个文件读取数据到2D numpy数组或列表的方法

debugcn 发表于 Dev

中子星

我有n许多文件，我想m从每个文件中特定行的列数中读取数据。我想以一个n按m列表或数组结尾，其中该列表或数组的每一行都是其中一个文件中的一行。有什么方法可以做到这一点？还是这是最好的方法？我可以定义m不同的列表或数组（每个都有相似的名称，可能只是按数字索引），但是处理几个不同的列表或数组似乎比仅一个二维的列表或数组复杂和复杂。

例如...

file_names = ( "file1" "file2" "file3" "file4" ) # so in this case n is 4
for name in file_names do:
    ###  somehow save the m columns as a row in an array or list
print resulting_list_or_array

有些可能相关的注意事项：m我想从文件中读取列不是第一次m在文件中列（比方说我需要跳过此skip_num列，我的确知道什么是n和m是。

另外，我要读取的行不是第一行，而是该row_num行。

索洛GP卡斯特罗

您可以使用内置模块glob和编写通用代码numpy.genfromtxt，如下面的示例所示，其中许多test*.txt文件的第二行将在最后建立一个二维数组：

from glob import glob

import numpy as np

def read(fname, skip_header=1):
    with open(fname) as f:
        for i in range(skip_header):
            f.next()
        return f.next()

a = np.genfromtxt((read(fname) for fname in glob('test*.txt')))

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。