擅长:python、mysql、java
<p>似乎需要返回匹配文件名的列表</p>
<p>您可以创建一个列表,将其累加到其中,然后返回:</p>
<pre><code>def get_path(input_folder):
retval = []
for file in os.listdir(input_folder):
if file.endswith(".txt"):
retval.append(os.path.join(input_folder, file))
return retval
</code></pre>
<p>或列表:</p>
<pre><code>def get_path(input_folder):
return [os.path.join(input_folder,file) for file in os.listdir(input_folder) if file.endswith(".txt")]
</code></pre>
<p>请注意,<code>glob</code>在本例中正好是这样做的(另外,在windows上,它还将匹配以<code>.TXT</code>(大写)结尾的文件,而第一个解决方案不匹配(除非在测试中插入<code>.lower()</code>):</p>
<pre><code>import glob, os
def get_path(input_folder):
return glob.glob(os.path.join(input_folder,"*.txt"))
</code></pre>