修改pandas dataframe对象时出现语法错误

Question

我有一个csv文件test.csv。

import pandas as pd
import numpy as np

在[0]

中

df = pd.read_csv('pandas/test.csv')

在[1]中

df.columns

出[0]

Index(['local_date', 'local_time','Latitude',
           'Longitude','SourceID','Vge','PacketClass'],
      dtype='object') 



df['Type'] = if MAN in df['Class'] return 'MAN' else return 'MAIN'

我收到语法错误^ SyntaxError: invalid syntax

在一行中如上所述在数据框内创建新列的方法是什么。

Answer 1

return语句只能在函数中使用。

重写：

df['Type'] = 'MAN' if MAN in df['Class'] else 'MAIN'