且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

如何在Python中从.txt文件加载特定行?

更新时间:2023-12-03 16:18:34

给出此文件:

1,2,3
4,5,6
7,8,9
10,11,12
13,14,15
16,17,18
19,20,21

您可以使用csv模块获取所需的np数组:

You can use the csv module to get the desired np array:

import csv
import numpy as np

desired=[1,3,5]
with open('/tmp/test.csv', 'r') as fin:
    reader=csv.reader(fin)
    result=[[int(s) for s in row] for i,row in enumerate(reader) if i in desired]

print(np.array(result))   

打印:

[[ 4  5  6]
 [10 11 12]
 [16 17 18]]