用python实现,从list中取指定几列组成新的list
时间: 2024-09-07 10:05:20 浏览: 69
在Python中,你可以使用列表推导式(List Comprehension)或者pandas库来从一个列表的子集中选取特定列。这里我们先假设你有一个二维列表(每个元素都是一个小列表),你想从中选择几个固定位置的元素。
如果你使用简单的列表推导式,可以这样做:
```python
def extract_columns(input_list, column_indices):
return [[row[i] for i in column_indices] for row in input_list]
# 示例
input_list = [['A', 'B', 'C'], ['D', 'E', 'F']]
column_indices = [0, 2] # 想选第一列和第三列
new_list = extract_columns(input_list, column_indices)
print(new_list) # 输出: [['A', 'C'], ['D', 'F']]
```
如果你的数据量大或者需要处理复杂的逻辑,使用pandas会更方便:
```python
import pandas as pd
def convert_to_pandas(input_list, columns):
df = pd.DataFrame(input_list)
return df[columns].tolist()
# 示例
input_list = ... # 同上
columns = ['A', 'C']
new_list = convert_to_pandas(input_list, columns)
print(new_list) # 输出:[['A', 'C'], ['D', 'F']]
```
阅读全文