用 Python 编写一个程序，将给定 DataFrame 中以"k"开头的城市和州名称存储到新的 CSV 文件中
pythonpandasserver side programmingprogramming
输入 −
假设，我们有一个包含城市和州列的 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
