我在 python 中有一個字典列表，其中包含浮點數和 np 數組的值。我想將它保存在熊貓數據框中，以便我可以為給定的一組參數繪制 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)但我收到這個錯誤ValueError: setting an array element with a sequence.因為我也得到了熊貓 df 的索引。有沒有一種有效的方法來做到這一點？
1 回答
犯罪嫌疑人X
TA貢獻2080條經驗 獲得超4個贊
dataframe您創建的結果如下所示：
當您要求: 時x, y = dfplot['xarr'], dfplot['yarr']，您要求的是列xarr&的內容，yarr它們都包含列表。因此，對于他們中的每一個，您都會得到一系列列表。
親眼看看：type(x)結果：
pandas.core.series.Series
您可以使用iloc訪問系列內容：
idx = 0 x = x_series.iloc[idx]
在您的示例中，您只有 1 條記錄，因此您可以簡單地使用：
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)
添加回答
舉報
0/150
提交
取消