diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
index 089a9296bcc4..0e2882d8c0fd 100644
--- a/.github/workflows/build_and_test.yml
+++ b/.github/workflows/build_and_test.yml
@@ -751,7 +751,9 @@ jobs:
         Rscript -e "install.packages(c('devtools', 'testthat', 'knitr', 'rmarkdown', 'markdown', 'e1071', 'roxygen2', 'ggplot2', 'mvtnorm', 'statmod'), repos='https://cloud.r-project.org/')"
         Rscript -e "devtools::install_version('pkgdown', version='2.0.1', repos='https://cloud.r-project.org')"
         Rscript -e "devtools::install_version('preferably', version='0.4', repos='https://cloud.r-project.org')"
-    - name: Install dependencies for documentation generation
+    # Should delete this section after SPARK 3.5 EOL.
+    - name: Install dependencies for documentation generation for branch-3.4, branch-3.5
+      if: inputs.branch == 'branch-3.4' || inputs.branch == 'branch-3.5'
       run: |
         python3.9 -m pip install 'sphinx==4.5.0' mkdocs 'pydata_sphinx_theme>=0.13' sphinx-copybutton nbsphinx numpydoc jinja2 markupsafe 'pyzmq<24.0.0'
         python3.9 -m pip install ipython_genutils # See SPARK-38517
@@ -760,6 +762,16 @@ jobs:
         gem install bundler -v 2.4.22
         cd docs
         bundle install
+    - name: Install dependencies for documentation generation
+      if: inputs.branch != 'branch-3.4' && inputs.branch != 'branch-3.5'
+      run: |
+        python3.9 -m pip install 'sphinx==5.0.0' mkdocs 'pydata_sphinx_theme>=0.13' sphinx-copybutton nbsphinx numpydoc jinja2 markupsafe 'pyzmq<24.0.0'
+        python3.9 -m pip install ipython_genutils # See SPARK-38517
+        python3.9 -m pip install sphinx_plotly_directive 'numpy>=1.20.0' pyarrow pandas 'plotly>=4.8'
+        python3.9 -m pip install 'docutils<0.18.0' # See SPARK-39421
+        gem install bundler -v 2.4.22
+        cd docs
+        bundle install
     - name: R linter
       run: ./dev/lint-r
     - name: Run documentation build