Jednym ze sposobów byłoby pobranie pierwszego wyniku przez asocjację, te indeksy asocjacyjne i tak są kolumnami. Zastosuj array_keys
aby je uzyskać, najpierw dodaj nagłówki, potem pierwszy pobrany wiersz, a następnie zapętl resztę.
// first set
$first_row = $STH->fetch(PDO::FETCH_ASSOC);
$headers = array_keys($first_row);
// $headers = array_map('ucfirst', $headers); // optional, capitalize first letter of headers
fputcsv($fp, $headers); // put the headers
fputcsv($fp, array_values($first_row)); // put the first row
while ($row = $STH->fetch(PDO::FETCH_NUM)) {
fputcsv($fp,$row); // push the rest
}
fclose($fp);