如何在R语言中扩展*运算符实现Python式字符串向量重复

在Python里，*运算符可以直接让列表重复指定次数，但R默认不支持字符串向量用*和数字相乘实现重复——不过你可以通过重载S3方法实现这个功能，具体步骤如下：

1. 为字符串类型重载*运算符 #

R的运算符本质是函数，*对应的是\`函数。我们可以为character类定义专属的方法，让它在遇到数字时调用rep()`实现重复：

# 定义字符串类型的*运算符方法
`*.character` <- function(e1, e2) {
  # 仅当右侧是数字时执行重复逻辑
  if (is.numeric(e2)) {
    rep(e1, times = e2)
  } else {
    # 非数字输入时保留原错误提示
    stop("non-numeric argument to binary operator")
  }
}

2. 测试效果 #

定义完方法后，就可以像Python那样使用*了：

a <- c('a', 'b', 'c')
a * 2
# 输出结果：
# [1] "a" "b" "c" "a" "b" "c"

3. 支持数字在前的写法（可选） #

如果想要支持2 * a这种数字在前的写法，还需要为numeric类补充*方法，确保数字和字符串向量相乘时也能触发重复逻辑：

# 定义数字类型的*运算符补充方法
`*.numeric` <- function(e1, e2) {
  if (is.character(e2)) {
    rep(e2, times = e1)
  } else {
    # 非字符串输入时调用原生的数字乘法，不破坏原有功能
    base::`*`(e1, e2)
  }
}

测试这个扩展：

2 * a
# 输出结果和之前一致：
# [1] "a" "b" "c" "a" "b" "c"

注意事项 #

• 这种重载是全局生效的，会影响当前R会话中所有字符串向量和数字的*运算，但因为R原本就不支持字符串和数字相乘（会报错），所以不会破坏原有正常功能。
• 如果只想在特定场景下使用这个功能，建议自定义一个新的S3类（比如pyvec），然后为该类定义*方法，避免全局修改。

内容的提问来源于stack exchange，提问作者Masoud