使用 Pandas DataFrame 模拟多维 Tensor 数据结构-Python教程-PHP中文网

使用 Pandas DataFrame 模拟多维 Tensor 数据结构

聖光之護

发布： 2025-09-05 15:43:17

原创

612人浏览过

使用 pandas dataframe 模拟多维 tensor 数据结构

本文旨在指导读者如何使用 Pandas DataFrame 模拟多维 Tensor 的数据结构，解决在 Pandas 中存储和操作类似 Tensor 的数据，并提供了一系列示例代码，展示如何进行数据访问、修改和聚合操作，帮助读者更有效地利用 Pandas 处理复杂的数据分析任务。

Pandas DataFrame 是一个强大的数据结构，虽然它本身不是一个真正的 Tensor，但可以通过巧妙地利用 DataFrame 的索引和数据操作功能，来模拟 Tensor 的行为。以下将详细介绍如何使用 DataFrame 来实现类似 Tensor 的数据存储、访问和计算。

1. DataFrame 的构建与索引设置

首先，需要将原始数据转换成 DataFrame。关键在于将 Tensor 的维度信息设置为 DataFrame 的索引。例如，对于一个包含 "time", "Col_A", "Col_B", "Col_C", "Col_D" 这五个维度的 Tensor，可以将这些维度设置为 DataFrame 的 MultiIndex。

import pandas as pd

# 示例数据
data = {
    'time': [123, 124, 125, 126, 127],
    'Col_A': ['A1', 'A5', 'A3', 'A2', 'A1'],
    'Col_B': ['B1', 'B3', 'B2', 'B1', 'B1'],
    'Col_C': ['C2', 'C7', 'C3', 'C2', 'C7'],
    'Col_D': ['D6', 'D1', 'D2', 'D6', 'D2'],
    'Price': [23.43, 14.63, 343.43, 43.43, 6.63]
}

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

# 设置 MultiIndex
df = df.set_index(['Col_A', 'Col_B', 'Col_C', 'Col_D']).sort_index()

print(df)

登录后复制

这段代码首先创建了一个 DataFrame，然后使用 set_index() 方法将 "Col_A", "Col_B", "Col_C", "Col_D" 列设置为索引。sort_index() 方法用于对索引进行排序，这有助于提高后续数据访问的效率。

2. 数据的访问与修改

设置好索引后，就可以使用 loc 属性来访问和修改 DataFrame 中的数据，就像操作 Tensor 一样。

# 访问特定位置的数据
price = df.loc[('A1', 'B1', 'C2', 'D6'), 'Price']
print(f"Price at (A1, B1, C2, D6): {price}")

# 修改特定位置的数据
df.loc[('A1', 'B1', 'C2', 'D6'), 'Price'] = 55.0
print(df)

# 同时修改多个值
df.loc[('A1', 'B1', 'C7', 'D2'), ['time', 'Price']] = [100, 99.0]
print(df)

登录后复制

df.loc[('A1', 'B1', 'C2', 'D6'), 'Price'] 用于访问索引为 ('A1', 'B1', 'C2', 'D6') 且列名为 'Price' 的数据。类似地，可以使用 df.loc[...] = ... 来修改数据。

3. 数据的聚合与计算

Pandas 提供了强大的聚合功能，可以方便地对 DataFrame 中的数据进行求和、平均等操作。可以使用 slice(None) 来表示通配符，从而实现对特定维度上的数据进行聚合。

即构数智人

即构数智人是由即构科技推出的AI虚拟数字人视频创作平台，支持数字人形象定制、短视频创作、数字人直播等。

查看详情

# 对特定维度求和
sum_price = df.loc[('A1', ['B1', 'B2', 'B3'], slice(None), ['D2', 'D6'])].sum()
print(sum_price)

登录后复制

df.loc[('A1', ['B1', 'B2', 'B3'], slice(None), ['D2', 'D6'])] 用于选择满足特定条件的数据，其中 slice(None) 表示选择该维度上的所有值。然后，使用 sum() 方法对选定的数据进行求和。

4. 扩展 DataFrame 的列

DataFrame 还可以方便地添加新的列，以存储更多的数据。

# 添加新列
df.loc[('A1', 'B1', 'C7', 'D2'), ['Address', 'Gender']] = ['1 Main St', 'male']
df.loc[('A4', 'B4', 'C4', 'D4')] = [130, 45.3, 'High St', 'female']
df.loc[('A1', 'B2', 'C3', 'D4'), ['time', 'Price', 'Address']] = [120, 51.4, '4 Brown St']

print(df)

登录后复制

这段代码演示了如何向 DataFrame 中添加新的列，并为特定位置的数据赋值。

5. 数据的查询与过滤

Pandas 提供了灵活的数据查询和过滤功能，可以根据条件选择满足特定条件的数据。

# 查询 Price 大于 50 的数据
df_filtered = df[df['Price'] > 50]
print(df_filtered)

# 查询 time 小于 125 且 Gender 为空的数据
df_filtered = df[(df['time'] < 125) & df['Gender'].isna()]
print(df_filtered)

登录后复制

这段代码演示了如何使用条件表达式来选择满足特定条件的数据。