Python-將字符串從文本文件存儲到變量中

Question

我想要一個腳本，該腳本將從輸入文件的每一行中的第一個單詞存儲到變量中。

當前代碼僅存儲姓氏。

def func(file): 
  global user 
  with open(file) as iFile: 
    for line in iFile: 
    string = line.split(',') 
    user = string[0]

func(sys.argv[1]) 
print user

文本文件輸入為：

string1,string2,string3
string1,string2,string3
string1,string2,string3

我希望變量user在文本文件的每一行上存儲所有string1。

Answer 1

縮進問題。

這將解決它。 我還更改了func，以便您不再使用全局函數，而是從列表中返回帶有用戶的生成器。

這是一種更好的做法，並且內存效率更高。

def func(file): 
  with open(file) as iFile: 
    for line in iFile: 
        string = line.split(',') 
        yield string[0]

for user in func(sys.argv[1]) 
    print user

Answer 2

我推薦這個

def func(path):
    with open(path) as _file:
        return [s.split(',')[0] for s in _file]

user = func(sys.argv[1])
print user

Answer 3

我會為此：

user = []
def func(file): 
  with open(file) as iFile: 
    for line in iFile: 
      user.append(line.split(',')[0])

func(sys.argv[1]) 
print user

它將在列表中存儲您將提供給該函數的每一行的所有“第一字符串”。 順便說一句，Matt PsyK的解決方案效率更高！