您知道如何将字符串列表拆分为不同的变量吗？

Question

我是python的初学者，我有一个可能很简单的问题。我有一个“file.txt”，原则上可以有 n 个字符串。

> file.txt

John
Rafa
Marta
... 
n

这被加载到程序中：

with open('/media/names.txt') as f:
    lines = f.read().splitlines()

现在，我从 csv 加载一个数据框，该数据框有一列（名称为“标识符”），其中包含很多名称。

Registration = pd.read_csv('/media/Registration.csv', 
    sep='\t', header=0)

目标是为每个变量分别找到 n 个字符串。例如，在这种情况下，我对列表中的第一个数据进行了处理：

names_1 = Registration[Registration['Identifier'].str.contains(lines[1])]
print(names_1)

仅保留以“John”为标识符的行。但是，我正在尝试创建 n 个数据框，因为“file.txt”列表中有项目。

names_1 = Registration[Registration['Identifier'].str.contains(lines[1])]

names_2 = Registration[Registration['Identifier'].str.contains(lines[2])]

names_3 = Registration[Registration['Identifier'].str.contains(lines[3])]

names_n = Registration[Registration['Identifier'].str.contains(lines[n])]

但我有点卡住了，我不知道如何执行此循环。谁来帮帮我？谢谢！

Answer 1

从理论上讲，您的问题的答案是局部变量存储在可通过函数 locals() 访问的字典中。因此，可以完全按照要求在循环中生成变量。

for i, line in enumerate(lines):
    locals()[f'names_{i}'] = Registration[Registration['Identifier'].str.contains(line)]

然而，你可以这样做并不意味着你应该这样做，以这种方式生成变量通常不是一个好主意。

问问自己，你将如何访问第 n 个变量？您正在走上一条使您的数据难以处理的道路。更好的方法是使用字典或列表等数据结构来轻松跟踪它。

names = []
for line in lines:
    names.append(Registration[Registration['Identifier'].str.contains(line)])

还要注意第一个索引是 0，而不是 1。

Answer 2

Python 列表索引从 0 开始。

试试这样的for循环：

for i in range(len(lines)):
  names = Registration[Registration['Identifier'].str.contains(lines[i])]

但是你需要保持names 的值。也许在一个列表中：

name_list = []
for i in range(len(lines)):
  names = Registration[Registration['Identifier'].str.contains(lines[i])]
  name_list.append(names)
print(name_list)

试试这个！享受编码！