读取douban_data.csv文件写入redis

需求:

  1. 把rank作为键
  2. value应该包含title,rating_num,comment_num,quote字段
  3. 获取键为99,对应的电影名称

读取的excel数据如下

ranktitlerating_numcomment_numquote
11aaaaaaaa
99nnnnnnnn

我的代码如下:

import redis
import pandas as pd
if __name__ == '__main__':
    # 读取excel数据
    df = pd.read_excel('F:\douban_data.xlsx')# 你的excel地址
    # 生成字典数据
    douban_data_dict = df.set_index(['rank'])[['title','rating_num','comment_num','quote']].to_dict('index')
    # 连接redis
    db = redis.Redis(host="你的redis主机", port=36379, db=3, password='你的redis密码', decode_responses=True)
    for rank in douban_data_dict:
        db.hmset(rank, douban_data_dict[rank])
    # 获取rank为99的电影名字
    print(db.hget(99, 'title'))

执行结果

img

望采纳,谢谢

使用pandas读取douban_data.csv。
然后将rank列作为keu,title,rating_num,comment_num,quote作为value:df.set_index(['rank'])[['title','rating_num','comment_num','quote']].to_dict()
redis,你直接连接你redis存就行了