编写一个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 次查看

启动你的职业生涯

完成课程获得认证

开始学习
广告