Numpy字符串操作 | rstrip()函数

简介

rstrip()函数是numpy字符串操作中的一种，它用于删除字符串末尾的指定字符，默认删除空格。该函数是一个字符串方法，适用于字符串数组或字符串中的单个字符串。

语法

numpy.char.rstrip(a, chars=None)

其中，a是要进行操作的字符串或字符串数组，chars是要删除的字符集合，默认删除空格。

返回值

函数返回一个与输入数组或字符串形状相同的字符串或字符串数组，其中所有末尾的给定字符被删除。

示例

import numpy as np

# 单个字符串的例子
s = 'hello '
print(np.char.rstrip(s))  # 'hello'
print(np.char.rstrip(s, ' '))  # 'hello' 等效于不加 chars 参数

# 字符串数组的例子
a = np.array(['hello ', 'world', '  ', '!', '  nice  '])
print(np.char.rstrip(a))  # ['hello' 'world' '' '!' '  nice']
print(np.char.rstrip(a, ' '))  # ['hello' 'world' '' '!' '  nice']
print(np.char.rstrip(a, '! '))  # ['hello' 'world' '  ' '' '  nice']

注意事项

• chars参数可以传入字符串或字符集合，多个字符需要以字符串形式传入，如chars='aeiou'。
• 与字符串方法相似，即便字符串数组中有某个元素末尾没有指定的字符，该元素也会被处理。如例子中的第二个元素'world'的末尾没有空格，但rstrip()函数仍然对其进行了处理。