📌  相关文章
📜  Python|获取每个子列表的第一个元素(1)

📅  最后修改于: 2023-12-03 15:19:22.431000             🧑  作者: Mango

Python | 获取每个子列表的第一个元素

在Python中,可以使用以下方法获取一个列表中每个子列表的第一个元素:

# 方法一:使用for循环
my_list = [[1,2,3],[4,5,6],[7,8,9]]
first_elements = []
for sub_list in my_list:
    first_elements.append(sub_list[0])
print(first_elements)
# Output: [1, 4, 7]

# 方法二:使用列表推导式
my_list = [[1,2,3],[4,5,6],[7,8,9]]
first_elements = [sub_list[0] for sub_list in my_list]
print(first_elements)
# Output: [1, 4, 7]

此外,还可以使用numpy库以及pandas库中的相关函数来获取每个子列表的第一个元素。

使用numpy:

import numpy as np
my_list = [[1,2,3],[4,5,6],[7,8,9]]
first_elements = np.array(my_list)[:,0]
print(first_elements)
# Output: [1 4 7]

使用pandas:

import pandas as pd
my_list = [[1,2,3],[4,5,6],[7,8,9]]
df = pd.DataFrame(my_list)
first_elements = df.iloc[:,0].tolist()
print(first_elements)
# Output: [1, 4, 7]

以上就是获取每个子列表的第一个元素的几种方法,您可以根据实际情况进行选择使用。