您的位置:

以Python新建DataFrame为中心

引言

在数据分析和数据处理过程中,最主要的操作之一就是数据表格的处理。而在Python语言中,pandas这个强大的库提供了DataFrame数据结构,给我们带来了很大的便利。

本文将会以Python新建DataFrame为中心,从多个方面对其进行详细的阐述。希望能够为读者在数据分析和数据处理过程中提供帮助。

正文

一、DataFrame是什么

DataFrame是一种二维的、表格形式的数据结构,类似于SQL中的表格或者Excel中的电子表格。它由pandas库提供,并且功能非常强大。

在DataFrame中,每列可以是不同的数据类型,比如整型、浮点型、字符串等。同时,我们还可以为每一列设置列名,并且可以对表中的数据进行访问和操作。

二、新建DataFrame

1、从csv文件中读取数据

代码示例:
import pandas as pd

df = pd.read_csv('data.csv')

在上面的代码中,我们通过pandas库中的read_csv函数,将data.csv文件中的数据读取到DataFrame中,并将结果赋值给变量df。

2、从字典中读取数据

代码示例:
import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],
        'Age': [28, 34, 29, 42],
        'Salary': [30000, 24000, 27000, 40000]}
df = pd.DataFrame(data)

print(df)

在上面的代码中,我们首先定义了一个字典data,它包含了三个键值对,分别表示姓名、年龄和薪水。接着,我们通过pandas库中的DataFrame函数,将字典data转换为DataFrame结构,并将结果赋值给变量df。最后,我们通过print函数将变量df输出。

三、DataFrame的基本操作

1、选择数据

我们可以通过loc和iloc两个属性来选择DataFrame中的数据。

在使用loc属性时,我们需要指定索引名称和列名,下面是一个例子:

代码示例:
import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],
        'Age': [28, 34, 29, 42],
        'Salary': [30000, 24000, 27000, 40000]}
df = pd.DataFrame(data)

print(df.loc[0, 'Name'])

print(df.loc[0:2, 'Name':'Age'])

在上面的代码中,我们首先按照上面的方式定义了一个DataFrame。然后,我们通过df.loc[0, 'Name']的方式选择了第一行第一列的数据,输出了Tom。接着,我们通过df.loc[0:2, 'Name':'Age']的方式选择了第一行到第三行,第一列到第二列的数据,并将结果输出。

在使用iloc属性时,我们需要指定索引和列号,下面是一个例子:

代码示例:
import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],
        'Age': [28, 34, 29, 42],
        'Salary': [30000, 24000, 27000, 40000]}
df = pd.DataFrame(data)

print(df.iloc[0, 0])

print(df.iloc[0:2, 0:2])

在上面的代码中,我们首先按照上面的方式定义了一个DataFrame。然后,我们通过df.iloc[0, 0]的方式选择了第一行第一列的数据,输出了Tom。接着,我们通过df.iloc[0:2, 0:2]的方式选择了第一行到第二行,第一列到第二列的数据,并将结果输出。

2、修改数据

我们可以通过直接赋值的方式,来修改DataFrame中的数据。下面是一个例子:

代码示例:
import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],
        'Age': [28, 34, 29, 42],
        'Salary': [30000, 24000, 27000, 40000]}
df = pd.DataFrame(data)

df.loc[0, 'Name'] = 'Jerry'

print(df)

在上面的代码中,我们首先按照上面的方式定义了一个DataFrame。然后,我们通过df.loc[0, 'Name']的方式选择了第一行第一列的数据,并将其修改为了Jerry。最后,我们通过print函数将变量df输出,查看数据是否被正确修改。

3、删除数据

我们可以通过drop函数来删除DataFrame中的数据。下面是一个例子:

代码示例:
import pandas as pd

data = {'Name': ['Tom', 'Jack', 'Steve', 'Ricky'],
        'Age': [28, 34, 29, 42],
        'Salary': [30000, 24000, 27000, 40000]}
df = pd.DataFrame(data)

df = df.drop(0)

print(df)

在上面的代码中,我们首先按照上面的方式定义了一个DataFrame。然后,我们通过df.drop(0)的方式删除了第一行数据。最后,我们通过print函数将变量df输出,查看数据是否被正确删除。

小结

本文主要针对Python新建DataFrame进行详细的阐述,并从多个方面对其进行了介绍。理解了DataFrame这个数据结构,我们可以更加轻松地进行数据分析和数据处理。