8个让Python代码效率翻倍的简单技巧

1

2

3

4

# 老掉牙的写法

temp=a

a=b

b=temp

1

2

3

4

5

# 运行一下试试：

a, b=3,5

print(f"交换前：a={a}, b={b}")

a, b=b, a# 这行就够了！

print(f"交换后：a={a}, b={b}")

1

2

3

4

5

# 老式写法

new_names=[]

fornameinfiles:

ifname.endswith('.txt'):

new_names.append(name.replace('.txt','.md'))

1

2

3

4

# 来测试一下：

files=['笔记1.txt','笔记2.txt','照片.jpg']

new_names=[name.replace('.txt','.md')fornameinfilesifname.endswith('.txt')]

print(f"处理后的文件名：{new_names}")

1

2

3

4

5

# 容易出错的写法

try:

value=user_data['age']

exceptKeyError:

value=0

1

2

3

4

# 实战体验：

user_data={'name':'小明'}

age=user_data.get('age',18)# 找不到就返回18

print(f"用户年龄：{age}")

1

2

3

4

5

6

7

# 各种魔法数字，看得头晕

ifuser_status==0:

print("未注册")

elifuser_status==1:

print("已注册")

elifuser_status==2:

print("已认证")

1

2

3

4

5

6

7

8

9

10

fromenumimportEnum

classUserStatus(Enum):

UNREGISTERED=0

REGISTERED=1

VERIFIED=2

# 实际使用起来很清晰

status=UserStatus.REGISTERED

print(f"用户状态：{status.name}")

1

2

3

4

5

6

7

8

ifgrade >=90:

level="A"

elifgrade >=80:

level="B"

elifgrade >=70:

level="C"

else:

level="D"

1

2

3

4

5

6

7

8

9

10

11

12

# 实战测试：

grade_map={

lambdax: x >=90:"A",

lambdax:80<=x <90:"B",

lambdax:70<=x <80:"C",

lambdax: x <70:"D"

}

defget_level(score):

returnnext(vfork, vingrade_map.items()ifk(score))

print(f"85分的等级：{get_level(85)}")

1

2

# 又臭又长的老写法

name=data['user']['profile']['name']

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

fromfunctoolsimportreduce

fromoperatorimportgetitem

defdeep_get(dictionary, keys, default=None):

returnreduce(lambdad, key: d.get(key, default)ifisinstance(d,dict)elsedefault,

keys.split('.'), dictionary)

# 实战一下：

data={

'user': {

'profile': {

'name':'张三',

'age':25

}

}

}

name=deep_get(data,'user.profile.name')

print(f"用户名：{name}")

1

2

3

f=open('data.txt','r')

# 处理文件

f.close()# 经常忘记关闭

1

2

3

4

# 实战演示：

withopen('demo.txt','w') as f:

f.write('这才是正确的打开方式！')

print("文件已自动关闭，无需担心!")

1

2

3

4

5

6

defcreate_user(name, age=None, city=None, hobby=None):

ifageisNone:

age=18

ifcityisNone:

city="北京"

# ...更多判断

1

2

3

4

5

6

7

8

9

10

11

12

13

14

defcreate_user(**kwargs):

defaults={

'age':18,

'city':'北京',

'hobby':'编程'

}

params={**defaults,**kwargs}

returnparams

# 测试一下：

user1=create_user(name='小明')

user2=create_user(name='小红', age=20)

print(f"用户1：{user1}")

print(f"用户2：{user2}")