提取一行的部分内容至一个numpy数组中

2024-04-18 07:35:34 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个文件,内容如下:

1 a 9 11 22 33 44 55
2 b 4 99 88 77 66 44
3 c 8 88 33 22 55 22

我想解析文件并将它们提取到2numpy数组中

letters = [a b c]
values  = [[11 22 33 44 55], [99 88 77 66 44], [88 33 22 55 22]]

我怎样才能以最有效的方式做到这一点。目前我正在使用for循环,如果我对numpy.loadtxt()也这样做的话。由于数据类型不同,我需要读取文件两次。你知道吗

我的代码是:

letters = np.loadtxt( train_file, usecols=[1], dtype=str)
values  = np.loadtxt( train_file, usecols=range(3,7))

Tags: 文件代码numpy内容fornp方式train
1条回答
网友
1楼 · 发布于 2024-04-18 07:35:34

一种方法是读取所需的列,然后将它们分开,如下所示:

data  = np.loadtxt( train_file, usecols=[1]+range(3,7))
letters=data[:,0]
values=np.delete(data,0,axis=1)

相关问题 更多 >