bằng len() trong Pandas dataframe

Question

Đây là giao diện của DataFrame tôi:

 
    StateAb GivenNm Surname     PartyNm PartyAb ElectedOrder 
35  WA  Joe BULLOCK Australian Labor Party  ALP    2 
36  WA Michaelia  CASH     Liberal  LP    3 
37  WA  Linda REYNOLDS     Liberal  LP    4 
38  WA  Wayne DROPULICH Australian Sports Party SPRT    5 
39  WA  Scott  LUDLAM   The Greens (WA)  GRN    6 

và tôi muốn liệt kê một danh sách các thượng nghị sĩ mà họ có chiều dài hơn 9 ký tự.

Vì vậy, tôi nghĩ rằng mã nên được như thế này:

df[len(df.Surname) >9] 

nhưng điều này làm nảy sinh một KeyError, nơi nào tôi đi sai?

Answer 1

Cách đúng để lọc một DataFrame dựa vào độ dài của chuỗi trong một cột là

df[df['Surname'].str.len() > 9] 

df['Surname'].str.len() tạo ra một loạt các độ dài cho cột họ và df[df['Surname'].str.len() > 9] lọc ra những cái nhỏ hơn hoặc bằng 9 Những gì bạn đã làm là kiểm tra độ dài của chính Series (số hàng nó có).

Answer 2

Hãy xem hàm python filter. Nó thực hiện chính xác những gì bạn muốn.

df = [ 
    {"Surname": "Bullock-ish"}, 
    {"Surname": "Cash"}, 
    {"Surname": "Reynolds"}, 
] 
longnames = list(filter(lambda s: len(s["Surname"]) > 9, df)) 
print(longnames) 

>>[{'Surname': 'Bullock-ish'}] 

Sytse