如何使用python,将pd矩阵A的每行的列标题为“起始索引”“结束索引”的元素(记为a,b)组成一个元素[a, b],并将所用行都执行这样的操作,存入B矩阵中
时间: 2024-09-26 19:05:18 浏览: 20
在Python中,你可以使用pandas库(pd)来实现这个需求。首先,假设`matrix A`是一个DataFrame,其中包含你需要提取的"起始索引"和"结束索引"。以下是如何操作:
```python
import pandas as pd
# 假设 matrix A 是这样的 DataFrame
matrix_A = pd.DataFrame({
'col1': [0, 5, 10], # 这里代表 "起始索引"
'col2': [1, 6, 11] # 这里代表 "结束索引"
})
# 创建一个新的空DataFrame B 来存储结果
matrix_B = []
# 遍历 matrix A 的每一行
for i, row in matrix_A.iterrows():
# 提取起始索引和结束索引,并打包成列表
a, b = row['col1'], row['col2']
# 将列表作为一个元素添加到新行中
new_row = [a, b]
# 将新行添加到 matrix_B
matrix_B.append(new_row)
# 将 list 转换回 DataFrame 并保存到 matrix_B
matrix_B = pd.DataFrame(matrix_B, columns=['起始索引', '结束索引'])
# 如果你想把结果直接赋值给现有的 dataframe,可以这样做
matrix_A['新列名'] = matrix_B.values.tolist()
```
如果你需要直接创建一个二维数组(list of lists),则无需最后一步转换,直接使用 `matrix_B = matrix_B.values.tolist()`。
阅读全文