python - 我们如何将 csv/xls 文件中两列的数据读入 2 个变量并将它们用于使用 python 的程序

Question

我们如何使用 Python 将 csv/xls 文件中两列的数据读取到 2 个变量中，以便将它们用于程序的后续部分？

通常我希望从 excel 文件中为下面的变量 a 和 b 选取数据，并给出表格中所有行的变量比率两列:

from difflib import SequenceMatcher

def similar(a, b):
    return SequenceMatcher(None,a ,b).ratio()

Answer 1

你可以使用 pandas 包来做到这一点:

import pandas as pd
df = pd.read_csv('filename.csv', usecols = ['a','b'])

它从您的文件中读取标题为“a”和“b”的列，并将它们放入数据框 df 中。有关更多选项，请参阅 read_csv . (而且还有一个类似的read_xls函数。)

然后您可以使用df['a'] 或df.a 访问df 列，例如:

SequenceMatcher(None, df['a'], df['b']).ratio()