根据列中的值和数字列表过滤行

时间:2020-10-20 10:44:33

标签: python pandas dataframe split data-analysis

我正在尝试通过将数字值列表与列中存在的数字匹配来从数据框中过滤行。问题在于该列包含数字和字符的组合。

例如:

mylist = [2012, 2045]

   Dept No
2012  - Management
2045  - Designing

我尝试了以下操作,但没有用

df_new = df[df['Dept No'].str.split(pat="-")[0].str.strip().isin(mylist)]

您能提出其他建议吗?

1 个答案:

答案 0 :(得分:0)

我们需要做的就是在代码中添加val j: java.util.Map[String, String] = JavaConversions.mapAsJavaMap(Map("a" -> "b")) ,它将起作用:

.str[0]