如果字符串少于 x 个字符，从字符串中删除最后一个单词的最高效计算方法是什么？

Question

我目前的解决方案是 x=3

a = "first one is"
b = "the second forever"

def fun(input):
  if input.split()[-1] < 3:
    return ' '.join( input.split()[0:-1])
  else:
    return input

fun(a)

“第一”

fun(b)

“永远的第二个”

有没有计算效率更高的东西？

Answer 1

你可以试试这个：

def fun2(input):
  s = input.rsplit(' ', 1)
  return s[0] if len(s[1]) < 3 else input

使用%timeit进行时间分析：

In [25]: def fun(input):
    ...:   if len(input.split()[-1]) < 3:
    ...:     return ' '.join( input.split()[0:-1])
    ...:   else:
    ...:     return input
    ...:

In [26]: def fun2(input):
    ...:   s = input.rsplit(' ', 1)
    ...:   return s[0] if len(s[1]) < 3 else input
    ...:

In [28]: fun(a), fun2(a)
Out[28]: ('first one', 'first one')

In [29]: %timeit fun(a)
433 ns ± 0.759 ns per loop (mean ± std. dev. of 7 runs, 1,000,000 loops each)

In [30]: %timeit fun2(a)
222 ns ± 1.04 ns per loop (mean ± std. dev. of 7 runs, 1,000,000 loops each)

fun2更快，因为rsplit稍快一些，并且避免了冗余计算。