Заменить 2-ю часть списка, возвращенную функцией в квадрате

Question

Как упомянуто выше, я хочу заменить 2-ю часть списка, возвращенного функцией, на 2-ю часть в квадрате.

n[s]:= {1*s,2*s};
ReplacePart[n[s],2->?^2]

Мне нужно, чтобы знак вопроса равнялся текущему возвращенному значению.Каков наиболее краткий способ сделать это с или без ReplacePart?

Answer 1

Возможно

n[s]:= {1*s,2*s}; 
n[s]/.{y_,z_}->{y,z^2}

, которое возвращает {s,4s^2}

Вы также можете написать это как

ReplaceAll[n[s],{y_,z_}->{y,z^2}]

Если список может иметь или не иметь более двухэлементы затем

ReplaceAll[n[s],{y_,z_,x___}->{y,z^2,x}]

сохранят любые дополнительные элементы без изменений