Bagaimana untuk Mengekstrak Lajur Tertentu daripada Fail CSV Menggunakan Python?

Memangkas Lajur CSV dengan Modul Python csv

Selalunya, adalah perlu untuk menghuraikan fail nilai dipisahkan koma (CSV) dan mengekstrak lajur data tertentu. Menggunakan modul csv Python, seseorang boleh menyelesaikan tugas ini dengan mudah.

Pengekstrakan Lajur

Untuk mengekstrak lajur tertentu, gunakan pembolehubah included_cols, yang menentukan indeks lajur menjadi disertakan. Setiap lajur dinomborkan bermula dari 0, dengan lajur pertama ialah 0. Contohnya, untuk menangkap lajur "ID," "Nama," "Zip" dan "Telefon", anda akan menulis included_cols = [1, 2, 6 , 7].

Lelaran dan Pencetakan

Untuk melelaran melalui baris dan mengekstrak lajur yang ditentukan, gunakan pemahaman senarai dalam gelung for. Untuk setiap baris, pembolehubah kandungan akan diisi dengan nilai daripada lajur yang ditentukan. Untuk memaparkan data yang diekstrak, gunakan cetakan(kandungan) dalam gelung.

included_cols = [1, 2, 6, 7]

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

Mengoptimumkan dengan Panda

Untuk pengendalian CSV yang lebih cekap, pertimbangkan untuk menggunakan modul panda . Pandas menyediakan alat yang mudah dan berkuasa untuk membaca dan memanipulasi fail CSV. Untuk membaca CSV dan menyimpan lajur tertentu ke dalam pembolehubah, hanya gunakan:

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

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