如何使用python从csv文件调用特定单词

2024-05-16 12:15:54 发布

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

假设我有一个.csv文件,有两列A, B,那么如果我调用列a第3行,那么如何打印列B第3行数据。请帮帮我,因为我是代码初学者

我只想调用一个输入值并从下一列打印相同的行值

这是我的代码:

如果消息中出现“Hello”:

     with open('data.csv', "r") as f:
        reader = csv.reader(f)
        chat = {}

        for row in reader:
            key = row[0]
            chat[key] = row[1:]
    try:
        gotreply = chat[msg]
    except KeyError:
        gotreply = 'null'

    print(gotreply)

     if gotreply == 'null':
        print("Sorry! I didn't understand. I'm still learning.")
    else:
        print("Input Again")

Tags: 文件csv数据key代码消息chatnull
3条回答

我不知道你到底在问什么,但你可以做

import pandas as pd

foo_df = pd.read_csv('files/example.csv')

if foo_df.A[2] == "c":
    print(foo_df.B[2])

我建议您熟悉Pandas作为解析数据的优雅方式

示例.csv:

A,B
a,b
b,c
c,d
d,e

其中A和B是列名

pandas.read_csv()将读取.csv并将其放入数据帧中Check out the docs试试看

import pandas as pd

foo_df = pd.read_csv('/path/to/example.csv')

print(foo_df)

应该回来

   A  B
0  a  b
1  b  c
2  c  d
3  d  e    

从这里开始,只需查看文档即可从数据框架中检索所需内容

我也是一个初学者,这是我在这里的第一篇文章,所以我会尽力帮助你

首先,我假设您的CSV文件有标题,我假设您的标题称为“A”和“B”

import csv

with open('test.csv', mode='r') as csv_file:
    csv_reader = csv.reader(csv_file)
    row_count = 1
    for i in csv_reader:
        if row_count == 4:
            print(i[1])  # prints second column of third row
        row_count += 1

也许有一种更有效的方法,但我对这个游戏也是新手

相关问题 更多 >