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

Chọn hàng chứa văn bản cụ thể với Pandas


Chọn hàng chứa text dùng str.contains().

File CSV CarRecords.csv.

Đọc CSV:

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

Chọn hàng chứa "BMW":

dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]

Ví dụ

import pandas as pd 
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv") 
print("DataFrame...\n", dataFrame) 
dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')] 
print("\nLấy hàng chứa BMW...\n", dataFrame)

Kết quả

DataFrame ... 
  Car  Place  UnitsSold 
0 Audi  Bangalore  80 
1 Porsche  Mumbai  110 
2 RollsRoyce  Pune  100 
3 BMW  Delhi  95 
4 Mercedes  Hyderabad  80 
5 Lamborghini  Chandigarh  80 
6 Audi  Mumbai  100 
7 Mercedes  Pune  120 
8 Lamborghini  Delhi  100 
Fetching rows with text BMW ... 
  Car  Place  UnitsSold 
3 BMW  Delhi  95