熊猫read_csv随机跳过具有特定条目的行



我有一个 csv 文件,我想在其中跳过随机百分比的行,但仅适用于其中一列具有特定条目的行。例如,我可能有一个包含以下内容的csv,我想跳过所有苹果条目的一定百分比:

| a  | b  | c  | d  | e      |
|----|----|----|----|--------|
0| 9  | 1  | 2  | 3  | apple  |
1| 8  | 4  | 5  | 6  | apple  |
2| 7  | 7  | 8  | 9  | apple  |
3| 6  | 10 | 11 | 12 | orange |
4| 5  | 13 | 14 | 15 | orange |
5| 4  | 16 | 17 | 18 | orange |
6| 3  | 19 | 20 | 21 | orange |
7| 2  | 22 | 23 | 24 | banana |
8| 1  | 25 | 26 | 27 | banana |
9| 0  | 28 | 29 | 30 | banana |

我知道我可以用类似的东西跳过整个文件中的行

df = pd.read_csv('fruit.csv', skiprows = lambda i: i>0 and random.random() > probability_value) 

我知道我也可以从数据框中选择苹果条目

df2 = df.loc[df['e'] == 'apple']

但是有没有一种简单的方法可以在导入 csv 时选择这些条目并应用跳过行,以便所有非"苹果"条目不受跳过行的影响?

您可以按如下方式进行操作,但我更喜欢在后期执行此操作。

df = pd.read_csv('fruit.csv').query("e != 'apple'")

最新更新