如何在多行.csv文件中选择特定零件? (Python3)

时间:2018-11-13 09:29:26

标签: python-3.x list csv sum

为了解释这个问题,我正在整理此示例。 如果我有一个.csv文件,如下所示:

alfa, bravo, Charlie, 1.31
Dragonball, manga, anime, 3.11
delta, Omega, cookie, 3.13
Dragonball, stan, lee, 1.13

我如何拿起以“七龙珠”作为第一部分的每一行的第四部分?如果该列表继续进行,我不知道第一行的哪几行都以“七龙珠”开头。

我尝试过:

list = []
for line in file:
    line = line.rstrip()
    part = line.split(",")
    if part[0] == Dragonball:
        list.append(part[3])

预期输出:

list = [3.11, 1.13]

1 个答案:

答案 0 :(得分:1)

您可以使用pandas轻松做到这一点:

import pandas as pd

df = pd.read_csv("path to your csv file")

print(list(df[df[0]=='Dragonball'][3]))

输出:

[3.11, 1.13]