Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python Pandas - Lặp và lấy hàng chứa văn bản mong muốn

Để lặp và lấy hàng chứa văn bản mong muốn, sử dụng phương thức itertuples() và find(). itertuples() lặp qua các hàng DataFrame.

Trước tiên, nhập thư viện cần thiết với bí danh:

import pandas as pd

File CSV trên Desktop như đường dẫn dưới:

C:\\Users\\amit_\\Desktop\\CarRecords.csv

Đọc file CSV và tạo Pandas DataFrame:

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv")

Lặp và lấy hàng chứa văn bản cụ thể. Chúng ta tìm cột Car với văn bản "Lamborghini":

for k in dataFrame.itertuples():
    if k[1].find('Lamborghini') != -1:
        print(k)

Ví dụ hoàn chỉnh

Mã nguồn đầy đủ:

import pandas as pd
# đọc file csv
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv")
print("DataFrame...\n", dataFrame)
# lặp và lấy hàng chứa văn bản cụ thể
# tìm cột Car với văn bản Lamborghini
for k in dataFrame.itertuples():
    if k[1].find('Lamborghini') != -1:
        print(k)

Kết quả

Chương trình sẽ tạo ra kết quả sau:

Pandas(Index=5, Car='Lamborghini', Place='Chandigarh', UnitsSold=80)
Pandas(Index=8, Car='Lamborghini', Place='Delhi', UnitsSold=100)