现有txt文件
Male Arched_Eyebrows Eyeglasses
0.jpg -1 1 1
1.jpg -1 1 1
2.jpg -1 -1 1
3.jpg -1 -1 1
同时提取Male和Eyeglasses列以及第0列
import pandas as pd
def main():
test = pd.read_table('train_label.txt', delimiter=' ')
male = test[['Male', 'Eyeglasses']]
print(male)
if __name__ == '__main__':
main()
结果:
第一列好像默认都会有
delimiter参数表示文件中的元素是用什么来进行分隔的。如果是,分隔,则
test = pd.read_table('train_label.txt', delimiter=',')
注意,原文件中Male前面是没有空格的,但是这样会出现分隔问题,所以我自己在Male前面加了空格