新人求解~~

机械蠕虫

def find_video(path):
    import os
    import os.path
    result = []
    list_top = list(os.walk(path))
    for each in list_top:
        for each_file in each[2]:
            if os.path.splitext(each_file)[1] == ('.mp4' or '.avi' or '.rmvb'):
                result.append(os.path.join(each[0], each[1], each_file))

    return result

path = input('请输入路径：')
print(find_video(path))
            
这是一个求出视频文件路径的程序，为啥输出结果总是[]空的，懵逼QAQ！            
            
            

机械蠕虫

路径输入反斜杠已经转义了~~

机械蠕虫

一X一

你可以在每一段for下面打印一下，看看每一步的输出是否达到了自己的要求
这个用来差错非常好用

机械蠕虫

一X一 发表于 2019-3-21 18:33
你可以在每一段for下面打印一下，看看每一步的输出是否达到了自己的要求
这个用来差错非常好用

问题就在 if 的条件os.path.splitext(each_file)[1]上，我打印了一下为[]，但是搞不懂为什么会是空的~~

塔利班

result.append(os.path.join(each[0], each_file))
还有你的逻辑只能判断.mp4

一X一

我看着像是你的逻辑出的问题，把‘or‘的条件分开写，因为你这种写法是查到的文件是否等于一个元组，很显然不等于嘛’，所以返回为空

机械蠕虫

一X一 发表于 2019-3-21 18:45
我看着像是你的逻辑出的问题，把‘or‘的条件分开写，因为你这种写法是查到的文件是否等于一个元组，很显然 ...

            if os.path.splitext(each_file)[1] == '.mp4':
                result.append(os.path.join(each[0], each_file))
               
            elif os.path.splitext(each_file)[1] == '.avi':
                result.append(os.path.join(each[0], each_file))
               
            elif os.path.splitext(each_file)[1] == '.rmvb':
                result.append(os.path.join(each[0], each_file))

这么写感觉有问题，太繁琐了，有更好的写法吗？

一X一

机械蠕虫 发表于 2019-3-21 18:58
if os.path.splitext(each_file)[1] == '.mp4':
                result.append(os.path.jo ...

换成成员操作符 ‘in’，是不是更简单一点

机械蠕虫

一X一 发表于 2019-3-21 19:00
换成成员操作符 ‘in’，是不是更简单一点

是的哦，谢谢

机械蠕虫

摸了半天，发现os.walk()是没有返回值的，list()只能用来在交互器中查看~~感谢6、9楼的纠正~