根据条件拆分列表（性别）

Question

我是一名Python初学者，刚刚学习如何编写函数。我有一份权重和性别列表，并试图根据性别标准将其分成两个新列表。使用for循环，我已经成功至今：

df = pd.read_csv('brainweight.csv') 
w = list(df['Weight']) 
s = list(df['Sex']) 

female_weight = [] 
male_weight = [] 

for sex, weight in zip (s, w): 
    if sex == 'f': 
     female_weight.append(weight) 
    else: 
     male_weight.append(weight) 

我应该如何修改这个for循环到一个函数，其中变量= M/F（性别）？

Answer 1

df = pd.read_csv('brainweight.csv') 
w = list(df['Weight']) 
s = list(df['Sex']) 
def list_by_gender(): 
    females = [] 
    males = [] 
    for sex, weight in zip (s, w): 
     if sex == 'f': 
      females.append(weight) 
     else: 
      males.append(weight) 
    return males,females 

male_list,female_list=list_by_gender()