Pandas 如何在for循环中向数据框中添加行

Pandas 如何在for循环中向数据框中添加行

在本文中,我们将介绍如何在for循环中将行添加到Pandas数据框。
通常,我们需要遍历一个数据集,并将每个数据点添加到数据框中。这种情况下,可以使用如下代码将每个数据点添加到数据框中:

import pandas as pd

# 创建一个空的数据框
df = pd.DataFrame(columns=['col1', 'col2'])

# 循环迭代
for i in range(10):
    col1 = i
    col2 = i * 2
    df = df.append({'col1': col1, 'col2': col2}, ignore_index=True)

print(df)
Python

上面的代码将创建一个空的数据框,并在for循环中迭代遍历,对于每个数据点,将数据点作为字典粘贴在数据框底部,并将ignore_index参数设置为True,以重置行号。当然,您可以根据您的数据设置列名。

如果您有一个数据列表,您也可以使用以下代码将其添加到数据框中:

import pandas as pd

# 创建一个空的数据框
df = pd.DataFrame(columns=['col1', 'col2'])

# 一个包含10个元素的列表
data = [[0, 0], [1, 2], [2, 4], [3, 6], [4, 8], [5, 10], [6, 12], [7, 14], [8, 16], [9, 18]]

# 将数据添加到数据框中
df = df.append(pd.DataFrame(data, columns=['col1', 'col2']), ignore_index=True)

print(df)
Python

在此示例中,我们创建了一个包含十个元素的列表,并将其添加到数据框中。我们还创建了一个包含列名的数据框,并将其与新数据拼接。

在for循环中使用Pandas数据框的一个更好的方法是使用 list 生成器并将其传入一个新的数据框中:

import pandas as pd

# 列名
cols = ['col1', 'col2']

# 用生成器创建数据
data = ((i, i*2) for i in range(10))

df = pd.DataFrame(data, columns=cols)

print(df)
Python

这段代码中,我们使用 PEP 289 引入的 generator 在for循环中迭代遍历数据,并创建一个新的数据框。

阅读更多:Pandas 教程

总结

在这篇文章中,我们介绍了在for循环中将数据行添加到 Pandas 数据框的三种方法,包括使用 for 循环和 append 方法,直接将列表添加到数据框中,以及使用 generator。希望这篇文章能够帮助您更好地理解如何使用 Pandas。

Python教程

Java教程

Web教程

数据库教程

图形图像教程

大数据教程

开发工具教程

计算机教程

登录

注册