編寫一個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

更新於:2021年2月24日

591 次檢視

啟動你的職業生涯

完成課程獲得認證

開始學習
廣告
© . All rights reserved.