python - 根据 pandas 中多列的条件删除随机 N 行

Question

df

    Text column  Title     Numbers column
0          abc   rom-com               1
1          xyz    comedy               2
2           hi   rom-com               4
3          jkl    murder               5
4          abc  thriller               2
and so on................

我想要什么:

我想要删除 5 行随机行(其中 Title 列具有值 rom-com)，并删除随机 6 行(其中 title 列具有值)值“谋杀”。

代码:

df1 = df.drop(df[df['Title'].str.contains('rom-com')].sample(5).index & /
[df['Title'].str.contains('murder')].sample(6).index)

错误:

AttributeError: 'list' object has no attribute 'sample'

上面的代码对于一个标题来说效果很好，但对于两个标题来说效果不佳。

df1 = df.drop(df[df['Title'].str.contains('rom-com')].sample(5).index \
#this alone works for both murder and rom-com separately.

但是两者一起我无法删除与多列中的值相对应的行。

Answer 1

可以使用Index.union :

df1 = df.drop(df[df['Title'].str.contains('rom-com')].sample(5).index.union(df[df['Title'].str.contains('murder')].sample(6).index))