markdown Python的常用功能代码段

Posted 2021-05-06

# 从命令行读取多个文件名，支持'\*'作为通配符
```python
import sys
from glob import glob

if len(sys.argv) >= 2:
    filelist = []
    for f in sys.argv[1:]:
        filelist.extend(glob(f))
    print(filelist)

    for f in filelist:
        # 将文件名传入你的处理函数
        process_func(f)
else:
    sys.exit('需要指定一个或多个文件名')
```

# 读取文件的文件名和扩展名
```python
import os

f = "a/b/c/d.txt"
name, ext = os.path.splitext(f)
print(name, ext)
print(os.path.dirname(f))
print(os.path.basename(f))
```
输出结果
```
a/b/c/d .txt
a/b/c
d.txt
```

# 写文件
```python
with open('a.txt', 'r') as f, open('b.txt', 'a') as out:
    for line in f:
        print(line)
    out.write('You’re still goin’ strong\n')
```

# 对列表去重，再按原顺序排序
```
ids = [1,4,3,3,4,2,3,4,5,6,1]
new_ids = list(set(ids))
new_ids.sort(key=ids.index)
```

# 判断变量是否为函数
```python
hasattr(f, '__call__')
```