Python中retrieveSong函数song列表未赋值问题求助

让我们一步步拆解你的代码问题，帮你解决这两个核心问题：song未赋值的报错，以及无法去除换行符的问题。

1. 为什么song变量未被赋值？ #

你遇到的第一个关键问题是文件指针耗尽：当你第一次遍历文件for line in file:统计行数后，文件的读取指针已经移动到了文件末尾。这时候第二个for line in file:循环根本不会执行任何迭代，自然永远触发不了count == randomIndex的条件，song也就从未被定义，最终导致返回时的报错。

另外还有一个索引范围错误：randint(0, length)会生成包含length的随机数，但文件实际只有length行，行的有效索引是从0到length-1（第一次循环count从0开始，最后count等于length，对应length行数据）。你现在的写法会随机到不存在的索引，即使指针没问题，也会找不到对应的行。

2. 为什么换行符\n没被去掉？ #

你的代码里item = str.strip(item)只是创建了一个去掉换行符的新字符串，但没有把这个新值存回原列表song里。原列表的元素还是原来带\n的字符串，所以看起来没生效。

修复后的完整代码 #

import random  # 别忘了导入random模块，否则randint会报错！

def retrieveSong(): 
    # Function that retrieves a random song from the file.
    with open("Songs.csv", "r") as file:  # 使用with语句自动管理文件关闭，更安全
        lines = file.readlines()  # 一次性读取所有行到列表，避免指针问题
    
    length = len(lines)
    if length == 0:  # 增加空文件判断，避免索引错误
        return []
    
    randomIndex = random.randint(0, length - 1)  # 修正索引范围
    song_line = lines[randomIndex].strip()  # 先去掉整行的换行符和前后空格
    song = song_line.split(",")
    
    # 处理每个元素的前后空格（比如艺术家名字可能有多余空格）
    song = [item.strip() for item in song]
    
    return song 
    # Returns the song name and artist, in a list where song[0] is the name and song[1] is the artist.

关键优化点解释 #

• 使用with open(...)语句：它会在代码块结束后自动关闭文件，不需要手动调用file.close()，避免忘记关闭文件的问题。
• 一次性读取所有行到lines列表：这样可以直接通过索引访问随机行，不需要两次遍历文件，从根源解决了指针耗尽的问题。
• 修正randint的范围：从0到length-1，确保取到的索引对应文件中存在的行。
• 使用列表推导式[item.strip() for item in song]：彻底去除每个元素的前后空格和换行符，真正修改列表里的元素。
• 增加空文件判断：如果Songs.csv是空的，返回空列表避免报错。

内容的提问来源于stack exchange，提问作者GamerAJ1025