Питон и БД - пните в верном направлении

Snowpark-DataFrame ленивый: все, что вы над ним пишете, превращается в единый SQL-план и исполняется прямо в кластере Snowflake. Поэтому самый прямой путь – описать, как меняется столбец, и вызвать update/merge/overwrite, не вытаскивая данные из базы.

Если изменение можно выразить обычным SQL, то банально:

 session.sql(""" 
    update my_schema.my_table 
       set price = price * 1.2 
     where category = 'books' 
""").collect() 

Никакой памяти на клиенте, выполняется распределённо.

Когда логика на чистом SQL не помещается и хочется Python, используем Snowpark-UDF. Пишем функцию, регистрируем как UDF, цепляем её в DataFrame и делаем update:

 from snowflake.snowpark.functions import col, udf 
 
@udf(name="normalize_phone", replace=True) 
def normalize_phone(phone: str) -> str: 
    import re 
    return re.sub(r'\D', '', phone)  # пример «сложной» питоновской логики 
 
df = session.table("contacts") 
df_to_update = df.with_column("phone", normalize_phone(col("phone"))) 
df_to_update.update({"phone": col("phone")})  # будет сгенерирован UPDATE ... SET phone = normalize_phone(phone) 

Метод update у DataFrame появился начиная с версии 1.1 Snowpark и генерирует обычный DML, так что всё опять же выполняется на стороне Snowflake.

Если трансформация затрагивает сразу много столбцов, проще создать временную таблицу и потом сделать MERGE. В коде это выглядит так:

 orig = session.table("sales") 
transformed = (orig 
               .with_column("net_amount", col("gross_amount") - col("discount")) 
               .with_column("updated_at", current_timestamp())) 
 
transformed.write.mode("overwrite").save_as_table("sales_tmp") 
 
session.sql(""" 
merge into sales t 
using sales_tmp s 
   on t.id = s.id 
when matched then update 
     set net_amount = s.net_amount, 
         updated_at = s.updated_at 
""").collect() 
 
session.sql("drop table sales_tmp").collect() 

При желании можно сделать SWAP вместо MERGE, если нужно полностью заменить таблицу без блокировки читателей: alter table sales_tmp swap with sales.

Читать всё в память, строить список Row и обратно грузить смысла нет, разве что вы тренируете модель в Pandas или нужен scikit-learn. Но для типового «обнови значения в таблице» правильный вектор такой: описываем изменение либо чистым UPDATE, либо через Snowpark-UDF, либо через цепочку DataFrame→save_as_table→MERGE, и даём Snowflake выполнить работу внутри себя. Вся магия DataSet из C# на самом деле как раз к этому и сводится, только Snowflake делает это эффективнее и без лишних раунд-трипов.

Если я правильно понимаю вопрос, то судя по доке Snowflake, это делается через SELECT ... FOR UPDATE:
https://docs.snowflake.com/en/sql-reference/constructs/for-update
Но это напрямую если запросами, а как это реализовано в используемых обёртках - не в курсе.

Нейросети пробовал?