首页 > 解决方案 > 如何使用 python 从我的 CSV 中捕获打印数据

问题描述

免责声明:我是第一次“程序员”,这是我的第一个项目。请耐心等待,因为我仍在学习术语以及与之相关的所有其他内容。

我正在尝试从我的 CSV 中捕获特定数据并将该数据输入到我的 python 脚本中以将其发送出去。(我正在制作一个 Twitter 机器人)。

这是我到目前为止的代码:

import csv

with open('data.csv') as csvDataFile:
    data = list(csv.reader(csvDataFile))

print(data[25][2])

这让我可以导入我的 CSV,然后选择我的推文引用所在的列和行。

这是我的数据的样子:

id_serial,id_season,quote,file_path,id_group,trigger_time
1,S1E01,"People say I am the best boss. They go ""god we've never worked at a place like this before. You're hilarious. And you get the best out of us.""",,001,09:00 AM EST
2,S1E01,I think this pretty much sums it up,/home/pi/Desktop/OfficialDundies/media/001.png,001,11:00 AM EST

我希望能够获取“打印的”报价并将其存储为变量,以便稍后回忆。理想情况下,我的机器人会抓取报价并检查“trigger_time”,看看是否该发推文了。

标签: pythoncsv

解决方案


研究 Pandas - 它通常比您自己编写的任何解析更容易和更有效地处理 CSV 数据。

您可以在此处找到安装说明,并此处找到简短教程。

您可以使用

import pandas as pd
df = pd.read_csv('data.csv')

然后

df['quote']

将是一个包含引号的系列(将其视为附有索引的值列表)。

要访问“我认为这个”报价,您可以使用

print (df['quote'].iloc[1])

你可能想要使用

df = df.set_index('id_serial')

这样索引就是您的 ID,然后您还可以使用

print (df['quote'].loc[2])

推荐阅读