python用np.loadtxt讀取csv文件

1. train.csv：

N_0000000435_00538.jpg
N_0000000436_00539.jpg
N_0000000437_00540.jpg
N_0000000356_00190.jpg
N_0000001506_00201.jpg
N_0000001507_00202.jpg
N_0000001939_00054.jpg

2. show_ids.py：

import numpy as np
ids = np.loadtxt('train.csv', delimiter='\t', dtype=np.str)
print([id[:-4] for id in ids])
print('\n')

ids=[id[:-4] for id in ids]
sample_id=ids[3]
print(sample_id)

#輸出結果：
['N_0000000435_00538', 'N_0000000436_00539', 'N_0000000437_00540', 'N_0000000356_00190', 'N_0000001506_00201', 'N_0000001507_00202', 'N_0000001939_00054']

N_0000000356_00190

3. 文件結構如下：

 

4. 運行結果：

python show_ids.py