dataframe常用操作

1. 数组转dataframe

import pandas as pd

input = [{"trainModelName": "东南 V6菱仕", "BaseModelName": "东南 V6菱仕", "label": "0"},
             {"trainModelName": "福特 经典全顺 ", "BaseModelName": "福特 经典全顺",
         "label": "0"}]


df = pd.DataFrame(input)
print(df)

2. dataframe转数组

df['properties'].values.tolist()

3. Object of type 'ndarray' is not JSON serializable

numpy的数组使用jsonify(arr)的时候报错， 因为json不认numpy的array。在后面加上tolist()。如 jsonify(arr.tolist())

4. 遍历dataframe

使用iterrows

for index, row in df.iterrows():
    brand = row['brand']

或者使用apply lambda组合：

df.apply(lambda row: generate_brand_sql(row['value'], row['letter']]), axis=1)

或者整个dataframe每一行执行方法：

df.apply(addflag, axis=1, args=(name,))          

def add flag(df, name):
    series = df.tolist() series.append(1)
    return pd.Series(series)