從數據框中刪除字符串長度大於某個數字的行，在某個字符（“，”）之后直到結束

Question

輸入：我有一個名稱包含 2 個值除以“，”的數據框

id name
1  xy,ab
2  cv,asdf
3  piy,bs

預期輸出：我想刪除“,”后名稱字符串長度大於 2 的行。

id name
1  xy,ab
3  piy,bs

我試過的代碼：

df = df[~df['name'].str.split().str.len().ge(2)]
df

此代碼僅刪除大於 2 的字符串長度，但我希望它發生在“，”之后。

Answer 1

您可以使用Series.str.match並傳遞regex ：

>>> df[df['name'].str.match('.*?,\w{0,2}$')]

   id    name
0   1   xy,ab
2   3  piy,bs

或者你可以用逗號分割值，取最后一個值，然后檢查長度是否小於或等於 2：

>>> df[df['name'].str.split(',').str[-1].str.len().le(2)]
   id    name
0   1   xy,ab
2   3  piy,bs