Pandas: cara membaca hanya baris tertentu dari file csv

Anda dapat menggunakan sintaks dasar berikut untuk hanya membaca baris tertentu dari file CSV ke dalam pandas DataFrame:

 #specify rows to import
specific_rows = [ 0 , 2 , 3 ]

#import specific rows from CSV into DataFrame
df = pd. read_csv (' my_data.csv ', skiprows = lambda x:x not in specific_rows)

Contoh khusus ini akan membaca baris pada posisi indeks 0, 2 dan 3 dari file CSV bernama my_data.csv ke dalam pandas DataFrame.

Contoh berikut menunjukkan cara menggunakan sintaksis ini dalam praktiknya.

Contoh: Hanya membaca baris tertentu dari file CSV di Pandas

Katakanlah kita memiliki file CSV berikut bernama basket_data.csv :

Jika kita menggunakan fungsi read_csv() , panda akan secara otomatis mengimpor setiap baris file CSV ke dalam DataFrame:

 import pandas as pd

#import all rows of CSV into DataFrame
df = pd. read_csv (' basketball_data.csv ')

#view DataFrame
print (df)

  team points rebounds
0 to 22 10
1 B 14 9
2 C 29 6
3 D 30 2

Namun, kita dapat menggunakan sintaks berikut untuk mengimpor hanya baris pada posisi indeks 0, 2 dan 3 dari file CSV ke dalam pandas DataFrame:

 import pandas as pd

#specify rows to import
specific_rows = [ 0 , 2 , 3 ]

#import specific rows from CSV into DataFrame
df = pd. read_csv (' basketball_data.csv ', skiprows = lambda x:x not in specific_rows)

#view DataFrame
print (df)

  team points rebounds
0 B 14 9
1 C 29 6

Perhatikan bahwa hanya baris pada posisi indeks 0, 2, dan 3 file CSV yang diimpor ke DataFrame.

Sintaks ini menggunakan argumen skiprows dan fungsi lambda untuk memberi tahu panda baris mana yang tidak boleh dilewati saat mengimpor file CSV.

Dalam contoh ini, kami memberi tahu panda untuk tidak mengabaikan baris di posisi indeks 0, 2, dan 3, namun mengabaikan semua baris lainnya saat mengimpor file CSV.

Catatan : Anda dapat menemukan dokumentasi lengkap fungsi pandas read_csv() di sini .

Sumber daya tambahan

Tutorial berikut menjelaskan cara melakukan tugas umum lainnya dengan Python:

Pandas: Cara melewati baris saat membaca file CSV
Pandas: Cara menambahkan data ke file CSV yang sudah ada
Pandas: Cara menentukan tipe saat mengimpor file CSV
Pandas: Cara mengatur nama kolom saat mengimpor file CSV