我是 Python 新手，在删除 csv 文件中不需要的行时遇到了问题。例如我有 3 列和很多行:
A B C
hi 1.0 5
hello 2.0 6
ok 3.0 7
我使用 numpy(而不是 csv)加载数据
import numpy as np
a= np.loadtxt('data.csv' , delimiter= ',' , skiprows= 1)
我想为第二列引入一个范围
b=np.arange(0, 2.1,0.1)
我不知道应该如何使用那段代码。
我想要的最终输出如下:
A B C
hi 1.0 5
hello 2.0 6
最后一行将被删除，因为我为第二列选择了一个范围，最高为 2.0。我不知道如何才能做到这一点。
最佳答案
尝试使用 Pandas:
import pandas as pd
a = pd.read_csv('data.csv', index_col=0) # column A will be the index.
a
B C
A
hi 1 5
hello 2 6
ok 3 7
对于 B 的每个值最多为 2 :
a[a.B <= 2]
B C
A
hi 1 5
hello 2 6
详细信息:
a.B
A
hi 1
hello 2
ok 3
Name: B, dtype: float64
a.B <= 2
A
hi True
hello True
ok False
Name: B, dtype: bool
关于python - 删除 CSV 文件中不需要的行，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/24864104/