Python DataFrame 添加一行|极客笔记

Python DataFrame 添加一行

在进行数据处理过程中，经常会遇到需要向DataFrame中添加新的一行数据的情况。本文将介绍如何使用Python中的pandas库来实现DataFrame添加一行的操作。

1. 创建DataFrame

首先，我们需要先创建一个DataFrame作为示例数据。下面是一个简单的示例：

import pandas as pd

data = {'Name': ['Alice', 'Bob', 'Charlie', 'David'],
        'Age': [25, 30, 35, 40],
        'City': ['New York', 'Los Angeles', 'Chicago', 'Houston']}

df = pd.DataFrame(data)
print(df)

运行上述代码，我们可以得到如下输出：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston

现在我们有一个包含姓名、年龄和城市信息的DataFrame。

2. 添加一行数据

接下来，我们将向上面的DataFrame中添加一行新的数据。我们可以通过构建一个字典来表示新的数据，然后将其添加到DataFrame中。

new_data = {'Name': 'Emily', 'Age': 28, 'City': 'San Francisco'}

df = df.append(new_data, ignore_index=True)
print(df)

运行上面的代码，我们可以看到DataFrame中成功添加了一行新的数据：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston
4    Emily   28  San Francisco

通过append方法，我们可以将新的数据添加到DataFrame的末尾，并且使用ignore_index=True参数可以自动为新数据分配递增的索引值。

3. 添加多行数据

如果我们想一次性添加多行数据，可以先创建一个包含多个字典的列表，然后将其一次性添加到DataFrame中。

new_data_list = [{'Name': 'Frank', 'Age': 45, 'City': 'Miami'},
                 {'Name': 'Grace', 'Age': 32, 'City': 'Seattle'}]

df = df.append(new_data_list, ignore_index=True)
print(df)

运行上述代码，我们可以看到DataFrame中成功添加了两行新的数据：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston
4    Emily   28  San Francisco
5    Frank   45        Miami
6    Grace   32      Seattle

4. 使用loc方法添加一行数据

除了使用append方法外，我们还可以使用loc方法来添加一行数据。这种方法更为灵活，可以按照指定的索引位置添加新的数据。

new_row = pd.Series(['Hannah', 27, 'Boston'], index=df.columns)
df = df.append(new_row, ignore_index=True)
print(df)

通过上述代码，我们可以看到DataFrame中成功添加了一行新的数据：

      Name  Age         City
0    Alice   25     New York
1      Bob   30  Los Angeles
2  Charlie   35      Chicago
3    David   40      Houston
4    Emily   28  San Francisco
5    Frank   45        Miami
6    Grace   32      Seattle
7   Hannah   27       Boston

使用pd.Series创建一个新的Series对象，其中指定了新数据的值和列名，并通过loc方法将其添加到DataFrame的末尾。