Question

我有一个学习者返回与日期对应的值列表。

我需要函数来返回数据帧以用于绘图目的。我已经创建了数据框，但现在我需要使用列表中的值填充数据框。这是我的代码：

learner.addEvidence(x,y_values.values)

y_prediction_list = learner.query(x) # this yields a plain old python list

y_prediction_df = pd.DataFrame(index=dates,columns="Y-Prediction")
y_prediction_df = ??

return y_prediction_df

Answer 1

您可以使用以下命令创建数据框：

y_prediction_df=pd.DataFrame({"Y-Prediction":y_prediction_list},index=dates)

Answer 2

我认为您可以使用DataFrame的参数data。

import pandas as pd

#test data
dates = ["2014-05-22 05:37:59", "2015-05-22 05:37:59","2016-05-22 05:37:59"]
y_prediction_list = [1,2,3]

y_prediction_df = pd.DataFrame(data=y_prediction_list, index=dates,columns=["Y-Prediction"])
print y_prediction_df
#                     Y-Prediction
#2014-05-22 05:37:59             1
#2015-05-22 05:37:59             2
#2016-05-22 05:37:59             3

print y_prediction_df.info()
#<class 'pandas.core.frame.DataFrame'>
#Index: 3 entries, 2014-05-22 05:37:59 to 2016-05-22 05:37:59
#Data columns (total 1 columns):
#Y-Prediction    3 non-null int64
#dtypes: int64(1)
#memory usage: 48.0+ bytes
#None

从列表

2 个答案: