Как выбрать конкретную часть в файле .csv в несколько строк?(Python3)

Question

Я делаю этот пример для объяснения проблемы.Если у меня есть файл .csv следующим образом:

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

Как я могу взять четвертую часть каждой строки, в которой в качестве первой части указывается «Dragonball»?Если список будет продолжен, и я не знаю, в каких строках в качестве первой части указан «Dragonball».

Я попытался:

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

Ожидаемый результат:

list = [3.11, 1.13]

Answer 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]