Bagaimana untuk Mengekstrak Data daripada Lajur Tertentu dalam Fail CSV dengan Modul CSV?

Mengekstrak Data daripada Lajur Tertentu dalam Fail CSV dengan Modul CSV

Dalam bekerja dengan fail CSV, selalunya perlu untuk mengekstrak data daripada lajur tertentu sahaja. Untuk mencapai matlamat ini menggunakan modul CSV, adalah penting untuk memahami cara pustaka mengendalikan pengambilan data.

Dalam sampel kod yang disediakan, percubaan untuk mengakses lajur tertentu adalah melalui sintaks baris[i], di mana i mewakili nombor lajur. Walau bagaimanapun, hasilnya tidak seperti yang dijangkakan kerana pendekatan yang salah.

Kaedah yang betul ialah memasukkan pernyataan cetakan dalam gelung for. Kod yang diperbetulkan berikut menunjukkan perkara ini:

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

Sekarang, kod yang diubah suai akan mencetak nilai daripada lajur yang diingini untuk setiap baris.

Walau bagaimanapun, adalah wajar untuk menyebut pendekatan alternatif menggunakan perpustakaan panda, yang memudahkan pengendalian fail CSV. Dengan panda, anda boleh membaca fail CSV dengan mudah dan mengekstrak lajur tertentu ke dalam pembolehubah:

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

Pustaka panda menyediakan set alat serba boleh untuk bekerja dengan data berstruktur, menjadikannya pilihan yang disyorkan untuk mengendalikan fail CSV dengan berkesan.

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