如何将pickle文件中的数据获取到pandas datafram中

2024-04-19 12:53:03 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在为一个班级做一个社交媒体情感分析。我已经收到了所有关于肯塔基德比的tweets,它们保存在pkl文件中长达2个月。

我的问题是:如何将所有这些pickle转储文件加载到数据帧中?

这是我的代码:

import sklearn as sk
import pandas as pd
import  got3

def daterange(start_date, end_date):
for n in range(int ((end_date - start_date).days)):
    yield start_date + timedelta(n)

start_date = date(2016, 3, 31)
end_date = date(2016, 6, 1)

dates = []

for single_date in daterange(start_date, end_date):
    dates.append(single_date.strftime("%Y-%m-%d"))

for i in range(len(dates)-1): 
    this_date = dates[i]
    tomorrow_date = dates[i+1]
    print("Getting tweets for " + tomorrow_date)
    tweetCriteria = got3.manager.TweetCriteria()
    tweetCriteria.setQuerySearch("Kentucky Derby")
    tweetCriteria.setQuerySearch("KYDerby")
    tweetCriteria.setSince(this_date)
    tweetCriteria.setUntil(tomorrow_date)
    Kentucky_Derby_tweets = got3.manager.TweetManager.getTweets(tweetCriteria)
    pkl.dump(Kentucky_Derby_tweets, open(tomorrow_date + ".pkl", "wb"))

Tags: 文件inimportfordatestarttweetstomorrow