2015-11-26 2 views
2

У меня есть ученик, который возвращает список значений, соответствующих датам.Создайте dataframe из списка

Мне нужна функция для возврата данных для целей построения графика. У меня создан DataFrame, но теперь мне нужно заполнить dataframe со значениями из списка. Вот мой код:

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 

ответ

1

вы можете просто создать dataframe с:

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

Я думаю, что можно использовать параметр data из DataFrame.

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 
Смежные вопросы