Fix sample_func in case where the token is an invalid Python variable name

[fsaad]

Example, the following variable cannot be rewritten in sample_func

X = Variable('X[0]')
sample_func(lambda ??: ??, 100)

One solution is to handle array variables by naming them X_at_10 in the lambda.