Bagaimana untuk Membaca Lajur Tertentu daripada Fail CSV dengan Cekap?

Membaca Lajur Terpilih dalam Fail CSV dengan Modul csv

Memahami Masalah

Membaca lajur tertentu daripada fail CSV boleh menjadi tugas biasa dalam pemprosesan data. Walau bagaimanapun, mengakses lajur mengikut nombornya, seperti yang dicuba dalam kod yang diberikan, mungkin membawa kepada hasil yang tidak dijangka.

Menangani Kesilapan

Kesilapan dalam kod yang disediakan terletak pada meletakkan pernyataan cetakan di luar untuk gelung. Akibatnya, hanya hasil lelaran terakhir dicetak, iaitu lajur terakhir fail CSV.

Pembetulan Kod

Untuk menyelesaikan isu ini, pernyataan cetakan harus dialihkan ke dalam gelung untuk mencetak lajur khusus setiap baris:

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Kelebihan Panda Modul

Walaupun kod yang dilaraskan boleh menyelesaikan tugas, ia tidak mempunyai kemudahan dan kecekapan modul panda. Pandas menyediakan penyelesaian yang elegan dan serba boleh untuk manipulasi fail CSV.

Pandas untuk Bacaan Lajur Terpilih

Menggunakan panda, membaca lajur tertentu daripada fail CSV menjadi mudah:

import pandas as pd

df = pd.read_csv(csv_file)

# Save column data to a variable
column_data = df['column_name']

Kod Ringkas

Penyelesaian Pandas memudahkan kod:

import pandas as pd

df = pd.read_csv(csv_file)

# Save specific columns as a list
specific_columns = ['ID', 'Name', 'Zip', 'Phone']
data = df[specific_columns].values.tolist()

Atas ialah kandungan terperinci Bagaimana untuk Membaca Lajur Tertentu daripada Fail CSV dengan Cekap?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!