我在 python 中有一個(gè)字典列表，其中包含浮點(diǎn)數(shù)和 np 數(shù)組的值。我想將它保存在熊貓數(shù)據(jù)框中，以便我可以為給定的一組參數(shù)繪制 xarr, yarr 。dic1 = {'p1':1, 'p2': 34, 'xarr' : np.array([1,2,3]) , 'yarr': np.array([4,4,6])}dic2 = {'p1':2, 'p2': 45 ,'xarr' : np.array([1,2,3]) , 'yarr': np.array([6,6,4])}listdic = [dic1, dic2]df = pd.DataFrame(listdic)dfplot= df[df['p1'] > 1]x, y = dfplot['xarr'], dfplot['yarr']plt.plot(x, y)但我收到這個(gè)錯(cuò)誤ValueError: setting an array element with a sequence.因?yàn)槲乙驳玫搅诵茇?df 的索引。有沒(méi)有一種有效的方法來(lái)做到這一點(diǎn)？
1 回答
犯罪嫌疑人X
TA貢獻(xiàn)2080條經(jīng)驗(yàn) 獲得超4個(gè)贊
dataframe您創(chuàng)建的結(jié)果如下所示：
當(dāng)您要求: 時(shí)x, y = dfplot['xarr'], dfplot['yarr']，您要求的是列xarr&的內(nèi)容，yarr它們都包含列表。因此，對(duì)于他們中的每一個(gè)，您都會(huì)得到一系列列表。
親眼看看：type(x)結(jié)果：
pandas.core.series.Series
您可以使用iloc訪問(wèn)系列內(nèi)容：
idx = 0 x = x_series.iloc[idx]
在您的示例中，您只有 1 條記錄，因此您可以簡(jiǎn)單地使用：
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
dic1 = {'p1':1, 'p2': 34, 'xarr' : np.array([1,2,3]) , 'yarr': np.array([4,4,6])}
dic2 = {'p1':2, 'p2': 45 ,'xarr' : np.array([1,2,3]) , 'yarr': np.array([6,6,4])}
listdic = [dic1, dic2]
df = pd.DataFrame(listdic)
dfplot= df[df['p1'] > 1]
x_series, y_series = dfplot['xarr'], dfplot['yarr']
x,y = x_series.iloc[0], y_series.iloc[0]
plt.plot(x, y)
添加回答
舉報(bào)
0/150
提交
取消