从一行 csv 文件中获取一条信息

Question

我有一个如下所示的 csv 文件：

name1, id1, email1, uID1
name2, id2, email2, uID2
name3, id3, email3, uID3
name4, id4, email4, uID4
name5, id5, email5, uID5
name6, id6, email6, uID6

我想从中获取一封随机电子邮件。例如。我想要email4，只有email4。我怎么读进去？我不想要 name4 id4 和 uID4，只需 email4。

注意：我正在编写一种方法来执行此操作，并且希望返回 email4 而不是打印它。

我已经看到很多关于如何获取整行或整列的信息，但没有看到如何获取一行的一部分。我该怎么做？

我已经查看并尝试了此线程上的所有选项：How can I get a specific field of a csv file? 但答案对我不起作用。所以新的解决方案或解决方案的修复会很棒！

这是我现在所在的位置：

num = random.randint(1,11)
    with open('Accounts_details.csv', 'rb') as f:
        reader = csv.reader(f)
        reader = list(reader)
        text = reader[num][2]
        print(text)

这会引发错误：

 reader = list(reader)
_csv.Error: iterator should return strings, not bytes (did you open the file in text mode?)

Answer 1

pandas 可能是最简单的方法。从您的帖子中不清楚您是想要特定字段还是随机字段，但两者都非常简单。

import pandas as pd
df = pd.read_csv(<your_file>)

print(df.iloc[4]['email'])

您可以随机生成 4 并将其用于索引，或者如果需要多个，则直接使用来自 pandas 的 random.sample。

Answer 2

您将其作为文本文件打开。

这个：

with open('Accounts_details.csv', 'rb') as f:  

应该是这样的：

with open('Accounts_details.csv', 'r') as f:  

Answer 3

您可以使用 numpy-library，完整示例：

import io
import numpy as np

test = """name1, id1, email1, uID1
name2, id2, email2, uID2
name3, id3, email3, uID3
name4, id4, email4, uID4
name5, id5, email5, uID5
name6, id6, email6, uID6"""

with open("test.txt", "w") as f:
    f.write(test)

data = np.genfromtxt("test.txt", delimiter="," ,dtype='unicode', autostrip=True)
# np.random.choice(data[:,2]) <-- random choice
data[:,2][4] # <--- index

更新：numpy和pandas的时间对比

%timeit np.genfromtxt("test.txt", delimiter="," ,dtype='unicode', autostrip=True)
# 1000 loops, best of 3: 404 µs per loop

%timeit pd.read_csv("test.txt", header=None, skipinitialspace=True)
1000 loops, best of 3: 954 µs per loop