Ausrichten der Achsen eingefügter Grafiken am Text im Dokument

Question

Ich versuche, ein pgfplot-Diagramm mit dem Textbereich des Dokuments auszurichten. Ich erstelle das Diagramm als eigenständiges PDF mit der scale only axisOption und width=\textwidthund bisher konnte ich es nur manuell mit \hspace*{-1.5cm}vor dem ausrichten \includegraphics{figure.pdf}. Das ist nicht nur etwas mühsam (ich habe ziemlich viele Diagramme!), sondern auch kaum präzise. Nachfolgend sehen Sie, was ich bisher tun kann:

Ist es möglich, dies stärker zu automatisieren, oder gibt es eine andere Möglichkeit, dies zu tun?

Hier ist der Code, den ich bisher habe:

% --- main.tex ---
\documentclass{article}
\usepackage{graphicx}
\usepackage{float}
\usepackage{showframe}

\begin{document} 

Lorem ipsum dolor sit amet, consectetur adipiscing elit. Aliquam cursus pharetra tristique. Curabitur sagittis libero sit amet ante fringilla, nec luctus neque laoreet. Morbi imperdiet convallis velit et elementum. 

\begin{figure}[h!]
\hspace*{-1.5cm}\includegraphics{figure.pdf}
\caption{Pellentesque molestie fermentum tellus ut posuere. Vestibulum euismod lectus quis augue commodo, imperdiet tincidunt felis viverra. Nulla facilisi.}
\end{figure}

Suspendisse id facilisis massa. Phasellus leo sapien, porttitor ac urna non, imperdiet maximus orci. Phasellus vel neque dui. Phasellus iaculis dictum dolor id iaculis. 

\end{document}

% --- figure.tex ---
\documentclass{standalone}

\usepackage{tikz}
\usepackage{pgfplots}

\begin{document}
\begin{tikzpicture}
\begin{axis}[
    xlabel=$x$,
    ylabel={$f(x)$},
    colorbar,
    width=\textwidth,
    height=2cm,
    scale only axis,
]
\addplot {x^2};
\end{axis}
\end{tikzpicture}
\end{document}