Python DataFrame 增加数据

Python DataFrame 增加数据

在数据处理和分析中，Python 中的 pandas 库是一个非常常用的工具。其中，DataFrame 是一个非常重要的数据结构，类似于 Excel 表格，可以用来存储和处理二维数据。在实际的工作中，我们经常需要向 DataFrame 中增加数据，本文将介绍在 Python 中如何向 DataFrame 中增加数据。

1. 使用 loc 方法增加一行数据

使用 loc 方法，可以在 DataFrame 的末尾增加一行数据。下面是一个示例代码：

import pandas as pd

data = {'A': [1, 2], 'B': [3, 4]}
df = pd.DataFrame(data)

new_data = {'A': 5, 'B': 6}
df.loc[len(df)] = new_data

print(df)

运行结果如下：

   A  B
0  1  3
1  2  4
2  5  6

2. 使用 append 方法增加多行数据

除了在末尾增加一行数据，我们还可以使用 append 方法一次性增加多行数据。下面是一个示例代码：

import pandas as pd

data = {'A': [1, 2], 'B': [3, 4]}
df = pd.DataFrame(data)

new_data = pd.DataFrame({'A': [5, 6], 'B': [7, 8]})
df = df.append(new_data, ignore_index=True)

print(df)

运行结果如下：

   A  B
0  1  3
1  2  4
2  5  7
3  6  8

3. 使用 loc 方法增加一列数据

使用 loc 方法，还可以在 DataFrame 中增加一列数据。下面是一个示例代码：

import pandas as pd

data = {'A': [1, 2], 'B': [3, 4]}
df = pd.DataFrame(data)

new_data = pd.Series([5, 6])
df.loc[:, 'C'] = new_data

print(df)

运行结果如下：

   A  B  C
0  1  3  5
1  2  4  6

4. 使用 assign 方法增加多列数据

除了增加一列数据，我们还可以使用 assign 方法一次性增加多列数据。下面是一个示例代码：

import pandas as pd

data = {'A': [1, 2], 'B': [3, 4]}
df = pd.DataFrame(data)

new_data = pd.DataFrame({'C': [5, 6], 'D': [7, 8]})
df = df.assign(**new_data)

print(df)

运行结果如下：

   A  B  C  D
0  1  3  5  7
1  2  4  6  8

结语

本文介绍了在 Python 中向 DataFrame 中增加数据的几种常用方法，包括使用 loc 方法增加一行或一列数据，使用 append 方法增加多行数据，以及使用 assign 方法增加多列数据。