问题描述
我有这个df
simbolo puntas ultimoPrecio
10 AE38 {'cantidadCompra': 79668.0,'precioCompra': 60... 6080.00
11 AL41C None 36.50
然后从“ puntas”列中的dict中获得第一个值。
p['Cant_Compra']=[x.get('cantidadCompra',0) for x in p['puntas']]
输出:
simbolo Cant_Compra ultimoPrecio
10 AE38 79668.0 6080.00
但是如果值是 None ,因为第11行出现下一条错误消息:
AttributeError:“ nonetype”对象没有属性“ get”
我该怎么办?
解决方法
import ast
from io import StringIO
# sample data
s = """simbolo|puntas|ultimoPrecio
AE38|{'cantidadCompra': 79668.0,'precioCompra': 60}|6080.00
AL41C|None|36.50"""
df = pd.read_csv(StringIO(s),sep='|')
df['puntas'] = df['puntas'].apply(ast.literal_eval)
# convert the dict column to a dataframe and assign the values to your column
df['Cant_Compra'] = pd.DataFrame(df['puntas'].to_dict()).T['cantidadCompra']
simbolo puntas ultimoPrecio \
0 AE38 {'cantidadCompra': 79668.0,'precioCompra': 60} 6080.0
1 AL41C None 36.5
Cant_Compra
0 79668
1 None
,
这有效:
def fun(x):
if x==None:
return None
return x['cantidadCompra']
p['Cant_Compra']=p['puntas'].apply(func=fun)