Python，获取多个列表并放入pd.Dataframe

基础概念

pandas 是一个强大的数据处理和分析库，它提供了 DataFrame 对象，用于存储和操作二维表格数据。DataFrame 可以从多种数据源（如列表、字典、SQL数据库等）创建。

类型

DataFrame 可以存储多种类型的数据，包括整数、浮点数、字符串、日期时间等。

应用场景

pandas 广泛应用于数据科学、机器学习、金融分析等领域。

示例代码

假设我们有三个列表，分别表示姓名、年龄和城市，我们希望将这些列表合并成一个 DataFrame。

import pandas as pd

# 定义三个列表
names = ['Alice', 'Bob', 'Charlie']
ages = [25, 30, 35]
cities = ['New York', 'Los Angeles', 'Chicago']

# 将列表转换为字典
data = {
    'Name': names,
    'Age': ages,
    'City': cities
}

# 创建 DataFrame
df = pd.DataFrame(data)

# 打印 DataFrame
print(df)

输出

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

遇到的问题及解决方法

问题：列表长度不一致

如果多个列表的长度不一致，创建 DataFrame 时会报错。

原因：DataFrame 要求每一列的数据长度一致。

解决方法：确保所有列表的长度一致，或者在创建 DataFrame 之前进行处理。

import pandas as pd

names = ['Alice', 'Bob']
ages = [25, 30, 35]
cities = ['New York', 'Los Angeles', 'Chicago']

# 确保所有列表长度一致
min_length = min(len(names), len(ages), len(cities))
names = names[:min_length]
ages = ages[:min_length]
cities = cities[:min_length]

data = {
    'Name': names,
    'Age': ages,
    'City': cities
}

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