python - 提供数据框第一列的值

Question

在其中一个代码片段中，作者将输入提供为：

variants = [ 'rs425277', 'rs1571149', 'rs1240707', 'rs1240708', 'rs873927', 'rs880051', 'rs1878745', 'rs2296716', 'rs2298217', 'rs2459994' ]

但是，我的值与 csv 格式的列之一相似。我想知道如何提供与上述示例类似的列之一作为输入？

提前致谢

标签： pythondataframecsvimport

Answer 1

首先，将您的 csv 作为 Pandas df 导入。

df = pd.read_csv('data.csv')  

然后，您可以从 pandas 数据框列中获取列表：

col_one_list = df['column_one'].tolist()