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