编写一个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
广告