在R中返回多个修改参数的最佳方法?

时间:2015-04-15 21:34:59

标签: r function arguments multiple-value

我尝试编写一个函数(或最好的R替代),给定一个 deck 向量和一个 hand 向量,它从第一个元素开始甲板并将其添加到手上。

# initialize the deck.  We don't care about suits, so modulus 13
deck <- sample(52, 52, replace = FALSE, prob = NULL) %% 13 + 1

# initilize the hand
hand <- vector(mode="numeric", length=0)

#deal a card from the deck to the hand.
# it's these two lines I'd like to put in something like a function 
#   and return the modified vectors
  phand <- c(hand, deck[1])
  deck <- deck[-1]

在像Python这样的东西中,你可以这样做:

def dealcard(deck,hand)

   # do something to deck and hand
   return deck, hand

deck,hand = dealcard(deck,hand)

看看第一个答案: modify variable within R function

  

@Dason表示有一些方法,但实际上 - 你不应该!

     

R的整个范式是&#34;通过值&#34;。 @Rory刚刚发布了处理它的正常方法 - 只返回修改后的值......

那么 R-thonic 最多的方法是什么?#34;修改函数的多个参数&#34;?

我可以看到用矩阵完成类似的事情,其中​​一列使用数字来指示行/卡的处理方式,但这会使代码更不直观。很容易想象一个功能:

score <- ScoreFunction(DealerHand)

VS:

score <- ScoreFunction(DeckMatrix, 1)

1 个答案:

答案 0 :(得分:1)

简短回答

您可以从函数返回list

简单示例

dealcard <- function(deck, hand) {
    # There needs to be a card to deal
    stopifnot(length(deck) > 0)

    # Add the card from the top of the deck to the hand
    hand <- c(hand, deck[1])

    # Remove the card from the deck
    deck <- deck[-1]

    # Create a named list from the modified inputs
    output <- list(deck=deck, hand=hand)

    return(output)
}

但请注意,此方法实际上并未修改全局环境中的deckhand对象。 (你可以使用<<-执行此操作,但在大多数情况下你不应该这样做。)所以你可以这样做:

# Initialize deck and hand as a list
# Full deck, empty hand
deck.hand <- list(deck=sample(52, 52, replace=FALSE), hand=NULL)

# Deal a card
deck.hand <- with(deck.hand, dealcard(deck, hand))

现在,如果您查看deck.hand,您会看到deckhand项目已得到适当更新。

更多涉及的例子

当同一个列表中有多个玩家时,可以轻松扩展以处理某个玩家。

# Initialize the game
# Three players, everyone starts with empty hands
game <- list(deck=sample(52, 52, replace=FALSE) %% 13 + 1,
             dealer=NULL, sally=NULL, john=NULL)


dealcard <- function(game, player, deck="deck") {
    # Make sure the player specified is in the game
    stopifnot(player %in% names(game) && deck %in% names(game))

    # There needs to be a card to deal
    stopifnot(length(game[[deck]]) > 0)

    # Give the next card to the specified player
    game[[player]] <- c(game[[player]], game[[deck]][1])

    # Remove the card from the deck
    game[[deck]] <- game[[deck]][-1]

    # Return the new state of the game
    return(game)
}

# Give everyone a card
game <- dealcard(game, "dealer")
game <- dealcard(game, "sally")
game <- dealcard(game, "john")

这使用语法list[["itemname"]]。列表项作为字符串传递。这相当于使用$提取项目,即list$itemname

这里我们传入整个列表,只修改它的必要部分,并返回整个修改后的列表。将原始列表设置为等于返回值就像修改列表一样。

现在,如果你查看game的内容,每个玩家将拥有一张牌,而牌组将丢失那些已分发的牌。

有趣的扩展

我对此有点了解。

想要连续向每位玩家交易五张牌吗?

players <- names(game)[names(game) != "deck"]

for (i in 1:5) {
    for (player in players) {
        game <- dealcard(game, player)
    }
}

再次查看game。每个玩家有5张牌,由于循环是以这种方式构建的,因此交易顺序模仿标准纸牌游戏。

您可以使用"deck"作为播放器,将播放器作为套牌,将卡片退回到套牌中,如下所示:

# John returns his first card to the deck
game <- dealcard(game, player="deck", deck="john")

鉴于以某种方式获得一手牌的功能,您可以轻松获得每个球员的得分。

# Example scoring
scorehand <- function(game, player) {
    return(sum(game[[player]]))
}

sapply(players, scorehand, game=game)

这将通过将scorehand()函数应用于向量players的每个元素来显示每个玩家的当前分数,该元素由玩家的名字组成。

TL; DR

使用列表。