📅  最后修改于: 2023-12-03 15:19:22.431000             🧑  作者: Mango
在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]
以上就是获取每个子列表的第一个元素的几种方法,您可以根据实际情况进行选择使用。