編寫一個Python程式,將給定DataFrame中以'k'開頭的城市和州名稱儲存到一個新的CSV檔案中。
輸入 −
假設我們有一個包含城市和州列的DataFrame,我們需要找到城市和州名稱以'k'開頭的記錄,並將它們儲存到另一個CSV檔案,如下所示:
City,State Kochi,Kerala
解決方案
為了解決這個問題,我們將遵循以下步驟。
定義一個DataFrame
檢查城市名稱是否以'k'開頭,如下所示:
df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
最後,將資料儲存到'CSV'檔案,如下所示:
df1.to_csv(‘test.csv’)
示例
讓我們來看下面的實現,以便更好地理解。
import pandas as pd
import random as r
data = { 'City': ['Chennai','Kochi','Kolkata'],'State':
['Tamilnad','Kerala','WestBengal']}
df = pd.DataFrame(data)
print("DataFrame is\n", df)
df1 = df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
df1.to_csv('test.csv')輸出
City,State Kochi,Kerala
廣告
資料結構
網路
關係資料庫管理系統 (RDBMS)
作業系統
Java
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP