Как в коллбэке array_map получить не только значение, но и ключ?

Передать ключи третьим аргументом!

Но учтите, что при использовании array_map с двумя и более массивами ключи не сохраняются, то есть результат всегда будет списком (индексным массивом).